diff --git a/egs/aishell2/data2vec_pretrain/run.sh b/egs/aishell2/data2vec_pretrain/run.sh index f07deb545..9acbbc787 100755 --- a/egs/aishell2/data2vec_pretrain/run.sh +++ b/egs/aishell2/data2vec_pretrain/run.sh @@ -42,7 +42,7 @@ train_set=train valid_set=dev_ios asr_config=conf/train_pretrain_transformer.yaml -model_dir="baseline_$(basename "${asr_config}" .yaml) _${lang}_${token_type}_${tag}" +model_dir="baseline_$(basename "${asr_config}" .yaml)_${lang}_${token_type}_${tag}" if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then echo "stage 0: Data preparation" diff --git a/funasr/build_utils/build_args.py b/funasr/build_utils/build_args.py index 632c13481..8a33063ba 100644 --- a/funasr/build_utils/build_args.py +++ b/funasr/build_utils/build_args.py @@ -54,6 +54,12 @@ def build_args(args, parser, extra_task_params): default=None, help="The number of input dimension of the feature", ) + task_parser.add_argument( + "--cmvn_file", + type=str_or_none, + default=None, + help="The path of cmvn file.", + ) elif args.task_name == "lm": from funasr.build_utils.build_lm_model import class_choices_list