fixed checkpoint->save_checkpoint

argonne-lcf · Feb 20, 2025 · 5c4df0b · 5c4df0b
1 parent d5e989c
commit 5c4df0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlio_benchmark/main.py b/dlio_benchmark/main.py
@@ -314,7 +314,7 @@ def _train(self, epoch):
         if self.do_checkpoint and (self.steps_between_checkpoints < 0) and (epoch == self.next_checkpoint_epoch):
             self.stats.end_block(epoch, block, block_step)
             self.stats.start_save_ckpt(epoch, block, overall_step)
-            self.checkpointing_mechanism.checkpoint(epoch, overall_step)
+            self.checkpointing_mechanism.save_checkpoint(epoch, overall_step)
             self.stats.end_save_ckpt(epoch, block)
             self.next_checkpoint_epoch += self.epochs_between_checkpoints
         return overall_step