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
1 changed files with 3 additions and 0 deletions

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()