From ac0111c196b141d88127b9c31313b953f2a9346a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 9 Sep 2019 10:55:47 -0400 Subject: [PATCH] Update multi_node_cluster_auto_slurm.py --- .../multi_node_examples/multi_node_cluster_auto_slurm.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/new_project_templates/multi_node_examples/multi_node_cluster_auto_slurm.py b/examples/new_project_templates/multi_node_examples/multi_node_cluster_auto_slurm.py index 8bf1c33506..1f2e1aad39 100644 --- a/examples/new_project_templates/multi_node_examples/multi_node_cluster_auto_slurm.py +++ b/examples/new_project_templates/multi_node_examples/multi_node_cluster_auto_slurm.py @@ -76,12 +76,11 @@ def main(hparams, cluster): # ------------------------ # 4 INIT TRAINER # ------------------------ - gpus = list(range(0, hparams.per_experiment_nb_gpus)) trainer = Trainer( experiment=exp, checkpoint_callback=checkpoint, early_stop_callback=early_stop, - gpus=gpus, + gpus=hparams.per_experiment_nb_gpus, nb_gpu_nodes=hyperparams.nb_gpu_nodes )