From a203e67affff5206cb525e0c9ce17d3493d0e1d4 Mon Sep 17 00:00:00 2001 From: songsenand Date: Mon, 6 Apr 2026 06:12:58 +0800 Subject: [PATCH] =?UTF-8?q?fix(trainer):=20=E4=BF=AE=E6=AD=A3=E6=80=BB?= =?UTF-8?q?=E8=AE=AD=E7=BB=83=E6=AD=A5=E6=95=B0=E8=AE=A1=E7=AE=97=E9=80=BB?= =?UTF-8?q?=E8=BE=91=E4=BB=A5=E6=94=AF=E6=8C=81=E5=A4=9A=E8=BD=AE=E8=AE=AD?= =?UTF-8?q?=E7=BB=83?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/model/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/model/trainer.py b/src/model/trainer.py index 02f8449..c855c0f 100644 --- 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,