diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 1eb10b23a7..38366d00dd 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -56,7 +56,8 @@ def main(hparams, cluster, results_dict): name=hyperparams.experiment_name, save_dir=hyperparams.test_tube_save_path, autosave=False, - description='test demo' + description='test demo', + version=cluster.hpc_exp_number ) exp.argparse(hparams)