update repo

This commit is contained in:
嘉渊 2023-05-25 11:07:18 +08:00
parent db788a1ff7
commit cb96f27364
2 changed files with 30 additions and 0 deletions

View File

@ -212,5 +212,19 @@ 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 conformer \
--dataset aishell2 \
--output_dir $exp_dir/exp/$model_dir \
--vocab_size $vocab_size \
--tag $tag
fi

View File

@ -220,4 +220,20 @@ 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 paraformer_bert \
--dataset aishell2 \
--output_dir $exp_dir/exp/$model_dir \
--vocab_size $vocab_size \
--nat _nat \
--tag $tag
fi