update

嘉渊

2023-07-28 6835108df0dfc56260892edb25d7d39618501e70

update

 egs/aishell2/data2vec_pretrain/run.sh

@@ -42,7 +42,7 @@
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"

 funasr/build_utils/build_args.py

@@ -54,6 +54,12 @@
            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