From 65ce10c255934e136288fe47d40ba7c87a83576a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 19:02:19 -0400 Subject: [PATCH] testing -1 gpu option --- tests/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index 3179b271..ed627fcc 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -195,7 +195,7 @@ def test_multi_gpu_model_dp(): max_nb_epochs=1, train_percent_check=0.1, val_percent_check=0.1, - gpus=[0, 1] + gpus='-1' ) run_gpu_model_test(trainer_options, model, hparams)