mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
added min accuracy to models test
This commit is contained in:
@@ -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