switched cpu amp order

2019-07-25 11:11:14 -04:00 · 2019-07-25 11:11:14 -04:00 · fffc09830f
parent aadf8e16aa
commit fffc09830f
1 changed files with 1 additions and 1 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -57,7 +57,7 @@ def test_amp_gpu_ddp_slurm_managed():

    # simulate setting slurm flags
    os.environ['MASTER_PORT'] = str(np.random.randint(12000, 19000, 1)[0])
-    os.environ['SLURM_LOCALID'] = 0
+    os.environ['SLURM_LOCALID'] = str(0)

    hparams = get_hparams()
    model = LightningTestModel(hparams)