From dbfcbbc0045cabb86ba4c0c26624b09b4e740e04 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 11:36:45 -0400 Subject: [PATCH] testing slurm ddp --- examples/new_project_templates/trainer_gpu_cluster_template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 99d8834c..72360f76 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -145,7 +145,7 @@ if __name__ == '__main__': # use default args root_dir = os.path.dirname(os.path.realpath(__file__)) log_dir = os.path.join(root_dir, 'pt_lightning_demo_logs') - checkpoint_dir = os.path.join(log_dir, 'model_weights') + checkpoint_dir = os.path.join(root_dir, 'model_weights') parent_parser = HyperOptArgumentParser(strategy='grid_search', add_help=False) # cluster args not defined inside the model