From f11eda857d82abc2603b053db446425ff00bf12e Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Mon, 8 Jul 2019 10:59:07 -0400
Subject: [PATCH] testing slurm ddp

---
 .../new_project_templates/trainer_gpu_cluster_template.py     | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py
index 56fedd30e0..22f4ea068b 100644
--- a/examples/new_project_templates/trainer_gpu_cluster_template.py
+++ b/examples/new_project_templates/trainer_gpu_cluster_template.py
@@ -50,7 +50,7 @@ def main(hparams, cluster, results_dict):
     log_dir = os.path.dirname(os.path.realpath(__file__))
     log_dir = os.path.join(log_dir, 'pt_lightning_demo_logs')
     exp = Experiment(
-        name='test_tube_exp',
+        name=hyperparams.tt_name,
         save_dir=log_dir,
         autosave=False,
         description='test demo'
@@ -83,7 +83,7 @@ def main(hparams, cluster, results_dict):
 
     # configure trainer
     trainer = Trainer(
-        experiment=hyperparams.tt_name,
+        experiment=exp,
         cluster=cluster,
         checkpoint_callback=checkpoint,
         early_stop_callback=early_stop,