added min accuracy to models test

This commit is contained in:
William Falcon
2019-07-24 08:53:59 -04:00
parent 5f810275c9
commit e62973dfd3
+3
View File
@@ -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()