From 53d5fea00aa9703ac56e675bbadef3824ac4164d Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 05 一月 2023 12:40:10 +0800
Subject: [PATCH] Merge pull request #27 from alibaba-damo-academy/dev
---
egs_modelscope/common/modelscope_utils/modelscope_infer.sh | 17 ++++++++---------
1 files changed, 8 insertions(+), 9 deletions(-)
diff --git a/egs_modelscope/common/modelscope_utils/modelscope_infer.sh b/egs_modelscope/common/modelscope_utils/modelscope_infer.sh
index 1a56dce..b9b3b6f 100755
--- a/egs_modelscope/common/modelscope_utils/modelscope_infer.sh
+++ b/egs_modelscope/common/modelscope_utils/modelscope_infer.sh
@@ -7,6 +7,7 @@
data_dir=
exp_dir=
model_name=
+model_revision=
inference_nj=32
gpuid_list="0,1,2,3"
njob=32
@@ -30,7 +31,7 @@
# download model from modelscope
python modelscope_utils/download_model.py \
- --model_name ${model_name}
+ --model_name ${model_name} --model_revision ${model_revision}
modelscope_dir=${HOME}/.cache/modelscope/hub/damo/${model_name}
@@ -48,14 +49,12 @@
fi
if "${use_lm}"; then
- cp ${modelscope_dir}/decode_asr_transformer.yaml ${modelscope_dir}/decode_asr_transformer.yaml.back
- cp ${modelscope_dir}/decode_asr_transformer_wav.yaml ${modelscope_dir}/decode_asr_transformer_wav.yaml.back
- sed -i "s#beam_size: [0-9]*#beam_size: `echo $beam_size`#g" ${modelscope_dir}/decode_asr_transformer.yaml
- sed -i "s#beam_size: [0-9]*#beam_size: `echo $beam_size`#g" ${modelscope_dir}/decode_asr_transformer_wav.yaml
- sed -i "s#lm_weight: 0.[0-9]*#lm_weight: `echo $lm_weight`#g" ${modelscope_dir}/decode_asr_transformer.yaml
- sed -i "s#lm_weight: 0.[0-9]*#lm_weight: `echo $lm_weight`#g" ${modelscope_dir}/decode_asr_transformer_wav.yaml
+ cp ${modelscope_dir}/decoding.yaml ${modelscope_dir}/decoding.yaml.back
+ sed -i "s#beam_size: [0-9]*#beam_size: `echo $beam_size`#g" ${modelscope_dir}/decoding.yaml
+ sed -i "s#lm_weight: 0.[0-9]*#lm_weight: `echo $lm_weight`#g" ${modelscope_dir}/decoding.yaml
fi
+ split_scps=
for n in $(seq "${inference_nj}"); do
split_scps+=" ${_logdir}/keys.${n}.scp"
done
@@ -67,6 +66,7 @@
${decode_cmd} --max-jobs-run "${inference_nj}" JOB=1:"${inference_nj}" "${_logdir}"/asr_inference.JOB.log \
python -m funasr.bin.modelscope_infer \
--model_name ${model_name} \
+ --model_revision ${model_revision} \
--wav_list ${_logdir}/keys.JOB.scp \
--output_file ${_logdir}/text.JOB \
--gpuid_list ${gpuid_list} \
@@ -85,6 +85,5 @@
done
if "${use_lm}"; then
- mv ${modelscope_dir}/decode_asr_transformer.yaml.back ${modelscope_dir}/decode_asr_transformer.yaml
- mv ${modelscope_dir}/decode_asr_transformer_wav.yaml.back ${modelscope_dir}/decode_asr_transformer_wav.yaml
+ mv ${modelscope_dir}/decoding.yaml.back ${modelscope_dir}/decoding.yaml
fi
--
Gitblit v1.9.1