funasr/bin/build_trainer.py
@@ -1,7 +1,7 @@ import os import yaml import logging def update_dct(fin_configs, root): if root == {}: @@ -55,7 +55,7 @@ scheduler_conf=None, specaug=None, specaug_conf=None, param_dict=None, mate_params=None, **kwargs): mode = modelscope_dict['mode'] args, ASRTask = parse_args(mode=mode) @@ -144,6 +144,10 @@ args.patience = None args.local_rank = local_rank args.distributed = distributed if mate_params is not None: for key, value in mate_params.items(): if hasattr(args, key): setattr(args, key, value) ASRTask.finetune_args = args return ASRTask