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')