From 44a10c64090ca2eed470bf7712257238c63df712 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 9 Apr 2020 14:33:46 -0400 Subject: [PATCH] workers warning not on windows --- 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 050c1d28c9d8a..7dad62234244c 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -386,7 +386,8 @@ def train(self): 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()