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

fix(model): 修正评估损失计算以避免除零错误

This commit is contained in:
songsenand 2026-02-26 01:00:16 +08:00
parent dc718cde5b
commit b0a4ce9ac8
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/trainer/model.py
View File

@ -517,7 +517,7 @@ class MoEModel(nn.Module):
if monitor: if monitor:
# 使用 eval_loss 作为监控指标 # 使用 eval_loss 作为监控指标
monitor.add_step( monitor.add_step(
global_step, {"loss": batch_loss_sum, "acc": acc} global_step, {"loss": batch_loss_sum / (eval_frequency if global_step > 0 else 1), "acc": acc}
) )
logger.info( logger.info(
f"step: {global_step}, eval_loss: {eval_loss:.4f}, acc: {acc:.4f}, 'batch_loss_sum': {batch_loss_sum / (eval_frequency if global_step > 0 else 1):.4f}, current_lr: {current_lr}" f"step: {global_step}, eval_loss: {eval_loss:.4f}, acc: {acc:.4f}, 'batch_loss_sum': {batch_loss_sum / (eval_frequency if global_step > 0 else 1):.4f}, current_lr: {current_lr}"