From e62973dfd30e28715fea6faaaf8701221ee84968 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 08:53:59 -0400 Subject: [PATCH] added min accuracy to models test --- tests/test_models.py | 3 +++ 1 file changed, 3 insertions(+) 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()