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 )