Update multi_node_own_slurm_script.py

2019-09-09 10:54:43 -04:00 · 2019-09-09 10:54:43 -04:00 · cbc619afa1
parent 3393086cb6
commit cbc619afa1
1 changed files with 1 additions and 1 deletions
--- a/examples/new_project_templates/multi_node_examples/multi_node_own_slurm_script.py
+++ b/examples/new_project_templates/multi_node_examples/multi_node_own_slurm_script.py
@ -41,7 +41,7 @@ def main(hparams):
    # ------------------------
    trainer = Trainer(
        experiment=exp,
-        gpus=[0, 1, 2, 3, 4, 5, 6, 7],
+        gpus=8,
        nb_gpu_nodes=2
    )