From 10e031a8434f4a44067177d5e4e0eca2b5ae197b Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Sat, 20 Jul 2019 09:17:20 -0400
Subject: [PATCH] removed logging

---
 pytorch_lightning/models/trainer.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index f153d6a9..16338e17 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -380,6 +380,9 @@ class Trainer(TrainerIO):
             # otherwise we launch the required number of processes
             try:
                 nb_slurm_tasks = int(os.environ['SLURM_NTASKS'])
+                print('x'*100)
+                print(f'NB_TASKS: {nb_slurm_tasks}')
+                print('x'*100)
                 nb_requested_gpus = len(self.data_parallel_device_ids)
                 is_slurm_managing_tasks = nb_slurm_tasks == nb_requested_gpus
             except Exception as e: