added min accuracy to models test
This commit is contained in:
parent
5f810275c9
commit
e62973dfd3
|
@ -27,14 +27,17 @@ def get_model():
|
|||
|
||||
return model
|
||||
|
||||
|
||||
def get_exp():
|
||||
exp = Experiment(debug=True)
|
||||
return exp
|
||||
|
||||
|
||||
def assert_ok_acc(trainer):
|
||||
# this model should get 0.80+ acc
|
||||
assert trainer.tng_tqdm_dic['val_acc'] > 0.80
|
||||
|
||||
|
||||
def test_cpu_model():
|
||||
model = get_model()
|
||||
|
||||
|
|
Loading…
Reference in New Issue