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

refactored model tests

Browse Source
This commit is contained in:
William Falcon committed 2019-07-24 12:04:11 -04:00
1 parent f50026c21f
commit de95179556
1 file changed
-2
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()