diff --git a/egs/aishell2/transformer/run.sh b/egs/aishell2/transformer/run.sh index bd46e6043..895e403ed 100755 --- a/egs/aishell2/transformer/run.sh +++ b/egs/aishell2/transformer/run.sh @@ -212,5 +212,18 @@ if [ ${stage} -le 5 ] && [ ${stop_stage} -ge 5 ]; then 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 transformer \ + --dataset aishell2 \ + --output_dir $exp_dir/exp/$model_dir \ + --vocab_size $vocab_size \ + --tag $tag +fi