嘉渊 2 lat temu
rodzic
commit
6835108df0

+ 1 - 1
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"

+ 6 - 0
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