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

2026-02-26 01:00:16 +08:00 · 2026-02-26 01:00:16 +08:00 · b0a4ce9ac8
parent dc718cde5b
commit b0a4ce9ac8
1 changed files with 1 additions and 1 deletions
--- a/src/trainer/model.py
+++ b/src/trainer/model.py
@ -517,7 +517,7 @@ class MoEModel(nn.Module):
                            if monitor:
                                # 使用 eval_loss 作为监控指标
                                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(
                                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}"