This commit is contained in:
嘉渊 2023-07-28 17:25:23 +08:00
parent f593c6fd83
commit 6835108df0
2 changed files with 7 additions and 1 deletions

View File

@ -42,7 +42,7 @@ train_set=train
valid_set=dev_ios valid_set=dev_ios
asr_config=conf/train_pretrain_transformer.yaml 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 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
echo "stage 0: Data preparation" echo "stage 0: Data preparation"

View File

@ -54,6 +54,12 @@ def build_args(args, parser, extra_task_params):
default=None, default=None,
help="The number of input dimension of the feature", 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": elif args.task_name == "lm":
from funasr.build_utils.build_lm_model import class_choices_list from funasr.build_utils.build_lm_model import class_choices_list