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

unify model test acc (#696)

Browse Source
This commit is contained in:
Jirka Borovec
2020-01-17 05:50:26 -05:00
committed by William Falcon
parent deb1581e26
commit bde549cb36
3 changed files with 11 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_cpu_models.py
+2 -2
View File
@@ -114,7 +114,7 @@ def test_running_test_after_fitting(tmpdir):
trainer.test()
# test we have good test accuracy
tutils.assert_ok_test_acc(trainer)
tutils.assert_ok_model_acc(trainer)
def test_running_test_without_val(tmpdir):
@@ -152,7 +152,7 @@ def test_running_test_without_val(tmpdir):
trainer.test()
# test we have good test accuracy
tutils.assert_ok_test_acc(trainer)
tutils.assert_ok_model_acc(trainer)
def test_single_gpu_batch_parse():