fix build_trainer param

This commit is contained in:
jmwang66 2023-02-09 11:25:57 +08:00
parent 4c26407a52
commit 3d54e6134c

View File

@ -49,7 +49,8 @@ def build_trainer(modelscope_dict,
scheduler_conf=None,
specaug=None,
specaug_conf=None,
param_dict=None):
param_dict=None,
**kwargs):
mode = modelscope_dict['mode']
args, ASRTask = parse_args(mode=mode)
# ddp related