mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-27 13:12:20 +08:00
added min accuracy to models test
This commit is contained in:
1 parent
5f810275c9
commit
e62973dfd3
1 file changed
+3
@@ -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()
|
||||
|
||||
|
||||
Reference in new issue
Block a user