diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index c149896448..2f2dfee2ff 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') + path = 'emv_' + os.environ('SLURM_CLUSTER_NAME') + 'group_' + os.environ['SLURM_*_PACK_GROUP_#'] os.makedirs(os.path.join(hparams.test_tube_save_path, path), exist_ok=True) # ------------------------