Fixes #2455

Author: williamFalcon

pytorch_lightning/callbacks/early_stopping.py

@@ -148,6 +148,10 @@ def _run_early_stopping_check(self, trainer, pl_module):
             if self.wait_count >= self.patience:
                 self.stopped_epoch = trainer.current_epoch
                 trainer.should_stop = True
+        print('-' * 100)
+        print('RUNNING EARLY STOP CHECK', trainer.global_rank)
+        print('stop:', trainer.should_stop)
+        print('-' * 100)
 
     def on_train_end(self, trainer, pl_module):
         if self.stopped_epoch > 0 and self.verbose > 0:

pytorch_lightning/trainer/training_loop.py

@@ -389,6 +389,9 @@ def train(self):
                                  f' ({self.min_epochs}) or minimum steps ({self.min_steps}) has'
                                  ' not been met. Training will continue...')
 
+            print('-' * 100)
+            print('SHUTTING DOWN', self.global_rank)
+            print('-' * 100)
             self.run_training_teardown()
 
         except KeyboardInterrupt: