From de9fc0587b6e4fa8b0884c66ae62b68022119177 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 26 Sep 2019 16:10:16 -0400 Subject: [PATCH] added docs for cluster grid search --- 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 62963e2c..f217bf18 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=2, + gpus=[0, 1], 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=2, + gpus=1, distributed_backend='ddp', use_amp=True )