diff --git a/tests/test_models.py b/tests/test_models.py index 87e9bd31..d34f3b4b 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -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()