update repo

This commit is contained in:
嘉渊 2023-05-25 10:54:53 +08:00
parent ba4aef18ed
commit cefa8ff17e

View File

@ -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 paraformer \
--model_name data2vec_finetune_paraformer \
--dataset aishell \
--output_dir $exp_dir/exp/$model_dir \
--vocab_size $vocab_size \
--tag $tag
fi