From 553223334f796b20f5c553b9604faef788d6f88b Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 14:11:48 -0400 Subject: [PATCH] using slurm flag to fine node nb --- 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 98cf87af..70813ceb 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -46,6 +46,7 @@ def main(hparams, cluster, results_dict): # 1 INIT LIGHTNING MODEL # ------------------------ print('loading model...') + print(os.environ['SLURM_SRUN_COMM_HOST']) model = LightningTemplateModel(hparams) print('model built')