From 3a28e52e9813f00a1f3a77e51962973ebc23002e Mon Sep 17 00:00:00 2001
From: ViperEkura <3081035982@qq.com>
Date: Thu, 28 May 2026 18:24:22 +0800
Subject: [PATCH] =?UTF-8?q?fix=20:=20start=5Fepoch/start=5Fbatch=20?=
 =?UTF-8?q?=E7=94=B1=E7=94=A8=E6=88=B7=E5=8F=82=E6=95=B0=E5=86=B3=E5=AE=9A?=
 =?UTF-8?q?=EF=BC=8C=E4=B8=8D=E5=86=8D=E8=A2=AB=20checkpoint=20=E8=A6=86?=
 =?UTF-8?q?=E7=9B=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 astrai/trainer/train_context.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/astrai/trainer/train_context.py b/astrai/trainer/train_context.py
index d80dae8..9fe0ae4 100644
--- 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: