update repo

This commit is contained in:
嘉渊 2023-05-25 10:58:45 +08:00
parent 5419042c7f
commit 325d28d1e8
3 changed files with 32 additions and 0 deletions

View File

@ -226,5 +226,6 @@ if [ ${stage} -le 6 ] && [ ${stop_stage} -ge 6 ]; then
--dataset aishell \
--output_dir $exp_dir/exp/$model_dir \
--vocab_size $vocab_size \
--nat _nat \
--tag $tag
fi

View File

@ -218,4 +218,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 aishell \
--output_dir $exp_dir/exp/$model_dir \
--vocab_size $vocab_size \
--nat _nat \
--tag $tag
fi

View File

@ -207,4 +207,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 transformer \
--dataset aishell \
--output_dir $exp_dir/exp/$model_dir \
--vocab_size $vocab_size \
--tag $tag
fi