feat : TrainConfig 支持 val_split 从训练集自动切分验证集

- val_split 比例从 dataset 中划出验证集，用 random_seed 固定随机切分 - 若 val_dataset 已显式设置则跳过自动切分
2026-06-02 20:33:40 +08:00
parent 0422d6d38e
commit 9fe2121743
2 changed files with 25 additions and 6 deletions
@@ -118,6 +118,12 @@ class TrainConfig(BaseConfig):
    val_dataset: Optional[Dataset] = field(
        default=None, metadata={"help": "Dataset for validation."}
    )
+    val_split: Optional[float] = field(
+        default=None,
+        metadata={
+            "help": "Ratio to split from training dataset for validation (e.g. 0.05). Ignored if val_dataset is set."
+        },
+    )
    val_step: int = field(
        default=1000,
        metadata={"help": "Number of optimizer steps between validation runs."},