From cbc619afa12e964749572c46509585bc6048beea Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 9 Sep 2019 10:54:43 -0400 Subject: [PATCH] Update multi_node_own_slurm_script.py --- .../multi_node_examples/multi_node_own_slurm_script.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git 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 index 7d4e286ab7..d8a383a90b 100644 --- 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 )