From 4f02791c6597c540697d4f453a7530415dba9b53 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 23 十二月 2022 19:26:07 +0800
Subject: [PATCH] Merge pull request #16 from alibaba-damo-academy/dev
---
egs_modelscope/common/modelscope_common_finetune.sh | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/egs_modelscope/common/modelscope_common_finetune.sh b/egs_modelscope/common/modelscope_common_finetune.sh
index 63143c8..8feacd0 100755
--- a/egs_modelscope/common/modelscope_common_finetune.sh
+++ b/egs_modelscope/common/modelscope_common_finetune.sh
@@ -7,7 +7,7 @@
gpu_num=2
count=1
gpu_inference=true # Whether to perform gpu decoding, set false for cpu decoding
-njob=4 # the number of jobs for each gpu
+njob=1 # the number of jobs for each gpu
train_cmd=utils/run.pl
infer_cmd=utils/run.pl
@@ -84,7 +84,7 @@
ngpu=$(echo $gpuid_list | awk -F "," '{print NF}')
if ${gpu_inference}; then
- inference_nj=$[${ngpu}*${njob}]
+ inference_nj=$njob
_ngpu=1
else
inference_nj=$njob
@@ -244,10 +244,10 @@
fi
${infer_cmd} --gpu "${_ngpu}" --max-jobs-run "${_nj}" JOB=1:"${_nj}" "${_logdir}"/asr_inference.JOB.log \
python -m funasr.bin.asr_inference_launch \
- --batch_size 1 \
+ --batch_size 64 \
--ngpu "${_ngpu}" \
--njob ${njob} \
- --gpuid_list ${gpuid_list} \
+ --gpuid_list ${gpuid_list:0:1} \
--data_path_and_name_and_type "${_data}/${scp},speech,${type}" \
--key_file "${_logdir}"/keys.JOB.scp \
--asr_train_config "${asr_exp}"/config.yaml \
--
Gitblit v1.9.1