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

refactored model tests

Browse Source
This commit is contained in:
William Falcon
2019-07-24 12:04:11 -04:00
parent f50026c21f
commit de95179556
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
-2
View File
@@ -135,7 +135,6 @@ def test_amp_gpu_dp():
# correct result and ok accuracy
assert result == 1, 'amp + gpu model failed to complete'
assert_ok_acc(trainer)
clear_save_dir()
@@ -169,7 +168,6 @@ def test_multi_gpu_model_ddp():
# correct result and ok accuracy
assert result == 1, 'multi-gpu ddp model failed to complete'
assert_ok_acc(trainer)
clear_save_dir()