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,