fix : start_epoch/start_batch 由用户参数决定，不再被 checkpoint 覆盖

2026-05-28 18:24:22 +08:00 · 2026-05-28 18:24:22 +08:00 · 3a28e52e98
parent e371908b54
commit 3a28e52e98
1 changed files with 2 additions and 2 deletions
--- a/astrai/trainer/train_context.py
+++ b/astrai/trainer/train_context.py
@ -80,8 +80,8 @@ class TrainContextBuilder:
                state_dict = load_model_weights(self._resume_dir)
            model.load_state_dict(state_dict, strict=False)
            if checkpoint is not None:
-                context.epoch = max(checkpoint.epoch, cfg.start_epoch)
-                context.iteration = max(checkpoint.iteration, cfg.start_batch)
+                context.epoch = cfg.start_epoch
+                context.iteration = cfg.start_batch
            context.checkpoint = checkpoint

        if cfg.lora is not None: