From cf04ff73e98330d8c901a2512ef7810f15f18f4c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 26 Sep 2019 16:10:51 -0400 Subject: [PATCH] undo test changes --- tests/test_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index f217bf18..62963e2c 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -339,7 +339,7 @@ def test_running_test_pretrained_model_dp(): val_percent_check=0.2, checkpoint_callback=checkpoint, experiment=exp, - gpus=[0, 1], + gpus=2, distributed_backend='dp' ) @@ -1247,7 +1247,7 @@ def test_amp_gpu_ddp_slurm_managed(): trainer_options = dict( show_progress_bar=True, max_nb_epochs=1, - gpus=1, + gpus=2, distributed_backend='ddp', use_amp=True )