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

added cpu + amp error

Browse Source
This commit is contained in:
William Falcon
2019-07-24 19:07:53 -04:00
parent efbd1a1c18
commit fcda19aa25
2 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
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():