fix : val_loss 默认改为 None，日志跳过空值；val_dataloader 补 Optional 注解

2026-06-13 14:24:13 +08:00
parent daf627a6de
commit 457e16ea3c
2 changed files with 9 additions and 5 deletions
@@ -31,8 +31,8 @@ class TrainContext:
    epoch: int = field(default=0)
    iteration: int = field(default=0)
    loss: float = field(default=0.0)
-    val_dataloader: DataLoader = field(default=None)
-    val_loss: float = field(default=0.0)
+    val_dataloader: Optional[DataLoader] = field(default=None)
+    val_loss: Optional[float] = field(default=None)

    world_size: int = field(default=1)
    rank: int = field(default=0)