Merge branch 'dev' into release-v2.3.3

DimensionSTP · May 17, 2024 · 722735c · 722735c
2 parents f92f035 + 9e48f9f
commit 722735c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/pipelines/pipeline.py b/src/pipelines/pipeline.py
@@ -100,7 +100,9 @@ def train(
         or config.strategy == "deepspeed_stage_3_offload"
     ):
         for epoch in range(config.epoch):
-            ckpt_path = f"{config.callbacks.model_checkpoint.dirpath}/epoch{epoch}.ckpt"
+            ckpt_path = (
+                f"{config.callbacks.model_checkpoint.dirpath}/epoch={epoch}.ckpt"
+            )
             if os.path.exists(ckpt_path) and os.path.isdir(ckpt_path):
                 convert_zero_checkpoint_to_fp32_state_dict(
                     ckpt_path,