From 2dec93f58895168ac3d426b0eb4a30e7c11dd480 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 9 Apr 2020 14:46:51 -0400 Subject: [PATCH] workers warning not on windows (#1433) --- pytorch_lightning/trainer/training_loop.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index 050c1d28c9..7dad622342 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -386,7 +386,8 @@ class TrainerTrainLoopMixin(ABC): self.run_training_teardown() except KeyboardInterrupt: - log.info('Detected KeyboardInterrupt, attempting graceful shutdown...') + if self.proc_rank == 0: + log.info('Detected KeyboardInterrupt, attempting graceful shutdown...') self.interrupted = True self.run_training_teardown()