diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 53aa99a7..356e8624 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -104,6 +104,7 @@ def main(hparams, cluster, results_dict): # ------------------------ trainer.fit(model) + def optimize_on_cluster(hyperparams): # enable cluster training # log all scripts to the test tube folder