From 3d54e6134c3884ada8a22fe8d1b15dc58edbd26f Mon Sep 17 00:00:00 2001 From: jmwang66 Date: Thu, 9 Feb 2023 11:25:57 +0800 Subject: [PATCH] fix build_trainer param --- funasr/bin/build_trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/funasr/bin/build_trainer.py b/funasr/bin/build_trainer.py index 5ef736a19..bb1d7a748 100644 --- a/funasr/bin/build_trainer.py +++ b/funasr/bin/build_trainer.py @@ -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