From a971ca515d21b67b980672d3e8a28f4deadd8b30 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 11:21:41 -0400 Subject: [PATCH] testing slurm ddp --- examples/new_project_templates/trainer_gpu_cluster_template.py | 1 + 1 file changed, 1 insertion(+) 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