From 8dab6d184a034ca86eafa644ea0d2100aadfe27d Mon Sep 17 00:00:00 2001
From: jmwang66 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 09 五月 2023 10:58:33 +0800
Subject: [PATCH] Merge pull request #473 from alibaba-damo-academy/dev_smohan

---
 funasr/runtime/python/utils/test_rtf.sh |   14 +++++---------
 1 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/funasr/runtime/python/utils/test_rtf.sh b/funasr/runtime/python/utils/test_rtf.sh
index 8ac2143..dcce6c4 100644
--- a/funasr/runtime/python/utils/test_rtf.sh
+++ b/funasr/runtime/python/utils/test_rtf.sh
@@ -10,8 +10,8 @@
 #:<<!
 model_name="damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch"
 backend="onnx" # "torch"
-quantize='True' # 'False'
-tag=${model_name}/${backend}_${quantize}
+quantize='true' # 'False'
+tag=${model_name}/${backend}_quantize_${quantize}
 !
 
 logs_outputs_dir=${export_root}/logs/${tag}/split$nj
@@ -19,18 +19,14 @@
 echo ${logs_outputs_dir}
 
 
-if [ $stage == 0 ];then
+if [ ${stage} -le 0 ];then
 
-  if [ $quantize == 'True' ];then
-    python -m funasr.export.export_model --model-name ${model_name} --export-dir ${export_root} --type ${backend} --quantize --audio_in ${scp}
-  else
-    python -m funasr.export.export_model --model-name ${model_name} --export-dir ${export_root} --type ${backend}
-  fi
+    python -m funasr.export.export_model --model-name ${model_name} --export-dir ${export_root} --type ${backend} --quantize ${quantize} --audio_in ${scp}
 
 fi
 
 
-if [ $stage -ge 1 ];then
+if [ ${stage} -le 1 ];then
 
 model_dir=${export_root}/${model_name}
 split_scps=""

--
Gitblit v1.9.1