From 2326aefc39d5070455eced7fa9acaadfb09eb154 Mon Sep 17 00:00:00 2001
From: Lizerui9926 <lzr265946@alibaba-inc.com>
Date: 星期五, 02 十二月 2022 22:43:13 +0800
Subject: [PATCH] update modelscope details
---
egs_modelscope/common/modelscope_common_finetune.sh | 9 +++++----
1 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/egs_modelscope/common/modelscope_common_finetune.sh b/egs_modelscope/common/modelscope_common_finetune.sh
index a43083f..39cd03e 100755
--- a/egs_modelscope/common/modelscope_common_finetune.sh
+++ b/egs_modelscope/common/modelscope_common_finetune.sh
@@ -11,7 +11,7 @@
train_cmd=utils/run.pl
# general configuration
-feats_dir="." #feature output dictionary, for large data
+feats_dir="../DATA" #feature output dictionary, for large data
exp_dir="."
lang=zh
dumpdir=dump/fbank
@@ -32,6 +32,7 @@
lfr_n=6
init_model_name=speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch # pre-trained model, download from modelscope during fine-tuning
+model_revision="v1.0.3" # please do not modify the model revision
cmvn_file=init_model/${init_model_name}/am.mvn
seg_file=init_model/${init_model_name}/seg_dict
vocab=init_model/${init_model_name}/tokens.txt
@@ -53,7 +54,7 @@
test_sets="dev test"
asr_config=conf/train_asr_paraformer_sanm_50e_16d_2048_512_lfr6.yaml
-init_param="init_model/${init_model_name}/${init_model_name}"
+init_param="init_model/${init_model_name}/model.pb"
inference_config=conf/decode_asr_transformer_noctc_1best.yaml
inference_asr_model=valid.acc.ave_10best.pth
@@ -61,7 +62,7 @@
. utils/parse_options.sh || exit 1;
# download model from modelscope
-python modelscope_utils/download_model.py --model_name ${init_model_name}
+python modelscope_utils/download_model.py --model_name ${init_model_name} --model_revision ${model_revision}
if [ ! -d ${HOME}/.cache/modelscope/hub/damo/${init_model_name} ]; then
echo "${HOME}/.cache/modelscope/hub/damo/${init_model_name} must exist"
@@ -158,7 +159,7 @@
world_size=$gpu_num # run on one machine
if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
# update asr train config.yaml
- python modelscope_utils/update_config.py --modelscope_config init_model/${init_model_name}/asr_train_config.yaml --finetune_config ${asr_config} --output_config init_model/${init_model_name}/asr_finetune_config.yaml
+ python modelscope_utils/update_config.py --modelscope_config init_model/${init_model_name}/finetune.yaml --finetune_config ${asr_config} --output_config init_model/${init_model_name}/asr_finetune_config.yaml
finetune_config=init_model/${init_model_name}/asr_finetune_config.yaml
mkdir -p ${exp_dir}/exp/${model_dir}
--
Gitblit v1.9.1