testing slurm ddp

2019-07-08 10:24:29 -04:00 · 2019-07-08 10:24:29 -04:00 · 861047672a
parent 65a0c98c5b
commit 861047672a
1 changed files with 1 additions and 1 deletions
--- a/examples/new_project_templates/trainer_gpu_cluster_template.py
+++ b/examples/new_project_templates/trainer_gpu_cluster_template.py
@ -181,7 +181,7 @@ if __name__ == '__main__':
        optimize_on_cluster(hyperparams)

    # RUN ON GPUS
-    if hyperparams.gpus is not None:
+    elif hyperparams.gpus is not None:
        # -1 means use all gpus
        # otherwise use the visible ones
        if hyperparams.gpus == '-1':