From a9acae3ed0c5605c5f6a28324854d1ec6cd9677c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 3 Jul 2019 18:17:12 -0400 Subject: [PATCH] checkpoint only on rank=0 now --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 21f0541d..91f1ef98 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -367,7 +367,7 @@ class Trainer(TrainerIO): self.batch_loss_value = 0 # accumulated grads # init progbar when requested - if self.progress_bar: + if self.progress_bar and self.proc_rank == 0: self.prog_bar = tqdm.tqdm(range(self.total_batches), position=self.process_position) for batch_nb, data_batch in enumerate(self.tng_dataloader):