scaled batch size

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