feat(trainer): 添加训练起始轮次和批次配置支持

2025-10-19 21:47:10 +08:00
parent 613edd7a14
commit 98efca7b9d
4 changed files with 27 additions and 5 deletions
@@ -34,6 +34,14 @@ class TrainConfig:
        default=4,
        metadata={"help": "Batch size for training."}
    )
+    start_epoch: int = field(
+        default=0,
+        metadata={"help": "Start epoch for training."}
+    )
+    start_batch: int = field(
+        default=0,
+        metadata={"help": "Start batch iteration for training."}
+    )
    checkpoint_interval: int = field(
        default=5000,
        metadata={"help": "Number of iterations between checkpoints."}