From 2d335c664c5e47e318781cb0f7f07649f60ac35c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 4 Oct 2019 13:07:22 -0400 Subject: [PATCH] Update multi_node_cluster_auto_slurm.py --- .../multi_node_examples/multi_node_cluster_auto_slurm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 133e79a5..7f201403 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 @@ -134,7 +134,7 @@ def optimize_on_cluster(hyperparams): cluster.add_slurm_cmd(cmd='constraint', value='volta32gb', comment='use 32gb gpus') cluster.add_slurm_cmd(cmd='partition', value=hyperparams.gpu_partition, - comment='use 32gb gpus') + comment='your cluster might need this argument') # run hopt # creates and submits jobs to slurm