From b4d85d299ffa8fbb14f62012fc1c9134530a38a4 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Sat, 8 May 2021 03:57:13 +0000
Subject: [PATCH] fix ckpt path

---
 examples/aishell/s0/run.sh         |  6 ++++--
 examples/aishell/s1/run.sh         |  6 ++++--
 examples/librispeech/s0/run.sh     | 13 +++++++------
 examples/librispeech/s1/.gitignore |  1 +
 examples/librispeech/s1/run.sh     |  7 ++++---
 examples/tiny/s0/run.sh            |  6 ++++--
 examples/tiny/s1/run.sh            |  6 ++++--
 7 files changed, 28 insertions(+), 17 deletions(-)

diff --git a/examples/aishell/s0/run.sh b/examples/aishell/s0/run.sh
index d4922748..d4961adb 100755
--- a/examples/aishell/s0/run.sh
+++ b/examples/aishell/s0/run.sh
@@ -6,12 +6,14 @@ gpus=0
 stage=0
 stop_stage=100
 conf_path=conf/deepspeech2.yaml
-ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
 avg_num=1
-avg_ckpt=avg_${avg_num}
 
 source ${MAIN_ROOT}/utils/parse_options.sh || exit 1;
 
+avg_ckpt=avg_${avg_num}
+ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
+echo "checkpoint name ${ckpt}"
+
 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
     # prepare data
     bash ./local/data.sh || exit -1
diff --git a/examples/aishell/s1/run.sh b/examples/aishell/s1/run.sh
index f0f66d53..01650229 100644
--- a/examples/aishell/s1/run.sh
+++ b/examples/aishell/s1/run.sh
@@ -5,12 +5,14 @@ source path.sh
 stage=0
 stop_stage=100
 conf_path=conf/conformer.yaml
-ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
 avg_num=20
-avg_ckpt=avg_${avg_num}
 
 source ${MAIN_ROOT}/utils/parse_options.sh || exit 1;
 
+avg_ckpt=avg_${avg_num}
+ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
+echo "checkpoint name ${ckpt}"
+
 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
     # prepare data
     bash ./local/data.sh || exit -1
diff --git a/examples/librispeech/s0/run.sh b/examples/librispeech/s0/run.sh
index 2ee577cd..472e6ebf 100755
--- a/examples/librispeech/s0/run.sh
+++ b/examples/librispeech/s0/run.sh
@@ -4,13 +4,14 @@ source path.sh
 
 stage=0
 stop_stage=100
-conf_path=conf/deepspeech2.yaml
-ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
-avg_num=1
-avg_ckpt=avg_${avg_num}
-
+conf_path=conf/transformer.yaml
+avg_num=30
 source ${MAIN_ROOT}/utils/parse_options.sh || exit 1;
 
+avg_ckpt=avg_${avg_num}
+ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
+echo "checkpoint name ${ckpt}"
+
 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
     # prepare data
     bash ./local/data.sh || exit -1
@@ -34,4 +35,4 @@ fi
 if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
     # export ckpt avg_n
     CUDA_VISIBLE_DEVICES= ./local/export.sh ${conf_path} exp/${ckpt}/checkpoints/${avg_ckpt} exp/${ckpt}/checkpoints/${avg_ckpt}.jit
-fi
\ No newline at end of file
+fi
diff --git a/examples/librispeech/s1/.gitignore b/examples/librispeech/s1/.gitignore
index 08728cdc..3423c574 100644
--- a/examples/librispeech/s1/.gitignore
+++ b/examples/librispeech/s1/.gitignore
@@ -1,3 +1,4 @@
 data
 exp
 log
+*.profile
diff --git a/examples/librispeech/s1/run.sh b/examples/librispeech/s1/run.sh
index 7cfc5686..472e6ebf 100755
--- a/examples/librispeech/s1/run.sh
+++ b/examples/librispeech/s1/run.sh
@@ -5,12 +5,13 @@ source path.sh
 stage=0
 stop_stage=100
 conf_path=conf/transformer.yaml
-ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
 avg_num=30
-avg_ckpt=avg_${avg_num}
-
 source ${MAIN_ROOT}/utils/parse_options.sh || exit 1;
 
+avg_ckpt=avg_${avg_num}
+ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
+echo "checkpoint name ${ckpt}"
+
 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
     # prepare data
     bash ./local/data.sh || exit -1
diff --git a/examples/tiny/s0/run.sh b/examples/tiny/s0/run.sh
index d4922748..d4961adb 100755
--- a/examples/tiny/s0/run.sh
+++ b/examples/tiny/s0/run.sh
@@ -6,12 +6,14 @@ gpus=0
 stage=0
 stop_stage=100
 conf_path=conf/deepspeech2.yaml
-ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
 avg_num=1
-avg_ckpt=avg_${avg_num}
 
 source ${MAIN_ROOT}/utils/parse_options.sh || exit 1;
 
+avg_ckpt=avg_${avg_num}
+ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
+echo "checkpoint name ${ckpt}"
+
 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
     # prepare data
     bash ./local/data.sh || exit -1
diff --git a/examples/tiny/s1/run.sh b/examples/tiny/s1/run.sh
index ffb1958c..f7e41a33 100644
--- a/examples/tiny/s1/run.sh
+++ b/examples/tiny/s1/run.sh
@@ -5,12 +5,14 @@ source path.sh
 stage=0
 stop_stage=100
 conf_path=conf/transformer.yaml
-ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
 avg_num=1
-avg_ckpt=avg_${avg_num}
 
 source ${MAIN_ROOT}/utils/parse_options.sh || exit 1;
 
+avg_ckpt=avg_${avg_num}
+ckpt=$(basename ${conf_path} | awk -F'.' '{print $1}')
+echo "checkpoint name ${ckpt}"
+
 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
     # prepare data
     bash ./local/data.sh || exit -1
-- 
GitLab