wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated reqs

Browse Source
This commit is contained in:
William Falcon
2019-07-24 09:39:43 -04:00
parent 1793d40b95
commit 6479f493ed
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+2 -1
View File
@@ -46,7 +46,8 @@ def clear_tt_dir():
def assert_ok_acc(trainer):
# this model should get 0.80+ acc
assert trainer.tng_tqdm_dic['val_acc'] > 0.70, "model failed to get expected 0.80 validation accuracy"
acc = trainer.tng_tqdm_dic['val_acc']
assert acc > 0.70, f'model failed to get expected 0.80 validation accuracy. Got: {acc}'
def test_cpu_model():