songsenand
/
SUimeModelTraner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

This commit is contained in:
songsenand 2026-04-06 06:12:58 +08:00
parent 6ad003133c
commit a203e67aff
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/model/trainer.py
View File

@ -1042,7 +1042,7 @@ def train(
model=model, model=model,
train_dataloader=train_dataloader, train_dataloader=train_dataloader,
eval_dataloader=eval_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, output_dir=output_dir,
num_epochs=num_epochs, num_epochs=num_epochs,
learning_rate=learning_rate, learning_rate=learning_rate,