update repo

This commit is contained in:
嘉渊 2023-05-25 11:07:34 +08:00
parent cb96f27364
commit 3300224541

View File

@ -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