From 5c41838d55d488827fd5e2258fd0714eff2a3014 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 13:37:21 -0400 Subject: [PATCH] added multi-node locked ip search --- examples/new_project_templates/trainer_gpu_cluster_template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 387060bb70..569d2ca408 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -42,7 +42,7 @@ def main(hparams, cluster, results_dict): :param hparams: :return: """ - path = 'emv_' + os.environ['SLURM_CLUSTER_NAME'] + 'group_' + os.environ['SLURM_*_PACK_GROUP_#'] + path = 'emv_' + os.environ['SLURM_CLUSTER_NAME'] os.makedirs(os.path.join(hparams.test_tube_save_path, path), exist_ok=True) # ------------------------