fix(train.py): 修复数据集加载时的参数传递问题

2025-10-06 16:44:02 +08:00 · 2025-10-06 16:44:02 +08:00 · 8c9e973179
parent fc98d9b7e6
commit 8c9e973179
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -66,7 +66,7 @@ def train(
    dataset = DatasetLoader.load(
        train_type=train_type,
        load_path=cache_files,
-        max_len=parameter.config.m_len
+        max_len=parameter.config.m_len,
        **kwargs
    )