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

added cpu + amp error

Browse Source
This commit is contained in:
William Falcon committed 2019-07-24 19:07:53 -04:00
1 parent efbd1a1c18
commit fcda19aa25
2 files changed
+3 -6
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/test_models.py
+2 -1
View File
@@ -40,7 +40,8 @@ def test_cpu_model_with_amp():
model, hparams = get_model()
run_gpu_model_test(trainer_options, model, hparams, on_gpu=False)
with pytest.raises(MisconfigurationException):
run_gpu_model_test(trainer_options, model, hparams, on_gpu=False)
def test_amp_gpu_ddp_slurm_managed():