fix(trainer): 修正总训练步数计算逻辑以支持多轮训练

2026-04-06 06:12:58 +08:00 · 2026-04-06 06:12:58 +08:00 · a203e67aff
parent 6ad003133c
commit a203e67aff
1 changed files with 1 additions and 1 deletions
--- a/src/model/trainer.py
+++ b/src/model/trainer.py
@ -1042,7 +1042,7 @@ def train(
        model=model,
        train_dataloader=train_dataloader,
        eval_dataloader=eval_dataloader,
-        total_steps=int(max_iter_length / batch_size),
+        total_steps=int(max_iter_length * num_epochs / batch_size),
        output_dir=output_dir,
        num_epochs=num_epochs,
        learning_rate=learning_rate,