From 4e67983f23a6ccfb45b1d421c1f824e26b7d9bbc Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Thu, 18 Jul 2019 17:05:09 -0400
Subject: [PATCH] added slurm no process warning

---
 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 3a03f4cc..a79af0d4 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -449,6 +449,9 @@ class Trainer(TrainerIO):
         except KeyError as e:
             node_rank = 0
 
+        print('-'*100)
+        print(gpu_nb)
+        print('-'*100)
         # recover original exp before went into process
         # init in write mode only on proc 0
         self.experiment.debug = self.proc_rank > 0