scaled batch size

2019-07-08 20:06:45 -04:00 · 2019-07-08 20:06:45 -04:00 · 7123dfeaf5
parent 9ee8f93483
commit 7123dfeaf5
1 changed files with 1 additions and 1 deletions
--- 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)