diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index fd9700fa96..aa4c52d411 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -693,6 +693,6 @@ class Trainer(TrainerIO): self.prog_bar.set_postfix(**tqdm_metrics) # model checkpointing - print('save callback...') if self.proc_rank == 0: + print('save callback...') self.checkpoint_callback.on_epoch_end(epoch=self.current_epoch, logs=self.__tng_tqdm_dic) \ No newline at end of file