From bc63bfe34b254bb9f752531c30845b211add3222 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 13:25:02 -0400 Subject: [PATCH] added multi-node locked ip search --- examples/new_project_templates/trainer_gpu_cluster_template.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 38366d00dd..c149896448 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -42,6 +42,9 @@ def main(hparams, cluster, results_dict): :param hparams: :return: """ + path = 'emv_' + os.environ('SLURM_CLUSTER_NAME') + os.makedirs(os.path.join(hparams.test_tube_save_path, path), exist_ok=True) + # ------------------------ # 1 INIT LIGHTNING MODEL # ------------------------