formatting

2019-08-04 13:06:25 -05:00 · 2019-08-04 13:06:25 -05:00 · 40e5f1b4c8
parent 181d69a727
commit 40e5f1b4c8
1 changed files with 1 additions and 1 deletions
--- 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)
+            self.checkpoint_callback.on_epoch_end(epoch=self.current_epoch, logs=self.__tng_tqdm_dic)