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

This commit is contained in:
2025-10-06 16:44:02 +08:00
parent fc98d9b7e6
commit 8c9e973179
+1 -1
View File
@@ -66,7 +66,7 @@ def train(
dataset = DatasetLoader.load( dataset = DatasetLoader.load(
train_type=train_type, train_type=train_type,
load_path=cache_files, load_path=cache_files,
max_len=parameter.config.m_len max_len=parameter.config.m_len,
**kwargs **kwargs
) )