diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index aba6219fd6..e60a0d95cc 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -885,4 +885,4 @@ class Trainer(TrainerIO): # model checkpointing if self.proc_rank == 0 and self.checkpoint_callback is not None: print('save callback...') - self.checkpoint_callback.on_epoch_end(epoch=self.current_epoch, logs=self.__tng_tqdm_dic) \ No newline at end of file + self.checkpoint_callback.on_epoch_end(epoch=self.current_epoch, logs=self.__tng_tqdm_dic)