testing -1 gpu option

2019-07-24 19:02:19 -04:00 · 2019-07-24 19:02:19 -04:00 · 65ce10c255
parent f58c83b399
commit 65ce10c255
1 changed files with 1 additions and 1 deletions
--- 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)