persist_ckpt option path args

intelligent-machine-learning · Jan 22, 2025 · 1a15252 · 1a15252
1 parent 262e63b
commit 1a15252
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlrover/python/training_event/predefined/trainer.py b/dlrover/python/training_event/predefined/trainer.py
@@ -118,7 +118,7 @@ def init(self, **kwargs) -> TrainInitSpan:
         """
         return self.custom_duration(TrainInitSpan, TrainerEventName.TRAIN_INIT.value, kwargs)
 
-    def persist_ckpt(self, global_step: int, path: str, size: Optional[int] = None, **kwargs):
+    def persist_ckpt(self, global_step: int, path: str = "", size: Optional[int] = None, **kwargs):
         """
         persist_ckpt event used in async checkpoint saving. stands for the training
         process is persisting