diff --git a/egs/aishell/data2vec_transformer_finetune/run.sh b/egs/aishell/data2vec_transformer_finetune/run.sh index 8e694e09f..27ba90c01 100755 --- a/egs/aishell/data2vec_transformer_finetune/run.sh +++ b/egs/aishell/data2vec_transformer_finetune/run.sh @@ -212,4 +212,19 @@ if [ ${stage} -le 5 ] && [ ${stop_stage} -ge 5 ]; then tail -n 3 ${_dir}/text.cer > ${_dir}/text.cer.txt cat ${_dir}/text.cer.txt done +fi + +# Prepare files for ModelScope fine-tuning and inference +if [ ${stage} -le 6 ] && [ ${stop_stage} -ge 6 ]; then + echo "stage 6: ModelScope Preparation" + cp ${feats_dir}/data/${train_set}/cmvn/am.mvn ${exp_dir}/exp/${model_dir}/am.mvn + vocab_size=$(cat ${token_list} | wc -l) + python utils/gen_modelscope_configuration.py \ + --am_model_name $inference_asr_model \ + --mode asr \ + --model_name data2vec_finetune_transformer \ + --dataset aishell \ + --output_dir $exp_dir/exp/$model_dir \ + --vocab_size $vocab_size \ + --tag $tag fi \ No newline at end of file