From 9b7311209e0a991e5afae2bc0e416cdcc78c18d5 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期三, 24 五月 2023 19:10:09 +0800
Subject: [PATCH] update repo

---
 egs/aishell/paraformer/run.sh |   11 +++++++++++
 1 files changed, 11 insertions(+), 0 deletions(-)

diff --git a/egs/aishell/paraformer/run.sh b/egs/aishell/paraformer/run.sh
index 373c119..0c6b73c 100755
--- a/egs/aishell/paraformer/run.sh
+++ b/egs/aishell/paraformer/run.sh
@@ -101,6 +101,7 @@
     utils/text2token.py -s 1 -n 1 --space "" ${feats_dir}/data/$train_set/text | cut -f 2- -d" " | tr " " "\n" \
         | sort | uniq | grep -a -v -e '^\s*$' | awk '{print $0}' >> ${token_list}
     echo "<unk>" >> ${token_list}
+    vocab_size=$(cat ${token_list} | wc -l)
 fi
 
 # LM Training Stage
@@ -213,4 +214,14 @@
 # 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
+    python utils/gen_modelscope_configuration.py \
+        --am_model_file $inference_asr_model \
+        --mode paraformer \
+        --model_name paraformer \
+        --model $model_dir \
+        --output_dir $exp_dir/exp/$model_dir \
+        --vocab_size $vocab_size \
+        --nar _nat \
+        --tag $tag
 fi
\ No newline at end of file

--
Gitblit v1.9.1