From 2c4177051b963953012650f66b6560f056ce6709 Mon Sep 17 00:00:00 2001
From: Jackwaterveg <87408988+Jackwaterveg@users.noreply.github.com>
Date: Thu, 16 Dec 2021 19:58:49 +0800
Subject: [PATCH] test=asr (#1140)

---
 examples/librispeech/asr2/run.sh | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/examples/librispeech/asr2/run.sh b/examples/librispeech/asr2/run.sh
index facaafcb..5b7596f2 100755
--- a/examples/librispeech/asr2/run.sh
+++ b/examples/librispeech/asr2/run.sh
@@ -7,9 +7,9 @@ set -e
 
 gpus=0,1,2,3,4,5,6,7
 stage=0
-stop_stage=100
+stop_stage=50
 conf_path=conf/transformer.yaml
-dict_path=data/bpe_unigram_5000_units.txt
+dict_path=lang_char/train_960_unigram5000_units.txt
 avg_num=10
 
 source ${MAIN_ROOT}/utils/parse_options.sh || exit 1;
@@ -48,11 +48,13 @@ if [ ${stage} -le 5 ] && [ ${stop_stage} -ge 5 ]; then
     CUDA_VISIBLE_DEVICES=0 ./local/align.sh ${conf_path} ${dict_path} exp/${ckpt}/checkpoints/${avg_ckpt} || exit -1
 fi
 
-# if [ ${stage} -le 6 ] && [ ${stop_stage} -ge 6 ]; then
-#     # export ckpt avg_n
-#     ./local/export.sh ${conf_path} exp/${ckpt}/checkpoints/${avg_ckpt} exp/${ckpt}/checkpoints/${avg_ckpt}.jit
-# fi
-
-if [ ${stage} -le 7 ] && [ ${stop_stage} -ge 7 ]; then
+if [ ${stage} -le 6 ] && [ ${stop_stage} -ge 6 ]; then
     ./local/cacu_perplexity.sh || exit -1
 fi
+
+if [ ${stage} -le 51 ] && [ ${stop_stage} -ge 51 ]; then
+    # export ckpt avg_n
+    ./local/export.sh ${conf_path} exp/${ckpt}/checkpoints/${avg_ckpt} exp/${ckpt}/checkpoints/${avg_ckpt}.jit
+fi
+
+
-- 
GitLab