diff --git a/examples/aishell/s0/run.sh b/examples/aishell/s0/run.sh index d49227483bd3ae071242b0b2ff04236154080f68..d4961adb2a36df578388d1575fa1f6f205fed55d 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 f0f66d53fa6fb50d005149c637213a38d0f1f2d6..016502298df6131c458edfdb82bf8993ee3044a8 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 2ee577cd58053b6f038f967375661bb4d29cb5a9..472e6ebfbd4d939bddb41771b5938c2b7595d6ff 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 08728cdc9c3e26ab42b8c941ef16fdb7fd0ae653..3423c574fb6b08ac1eab2f5d1f638cfd1942199d 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 7cfc5686f0ba081355a9a1fa532bc8d345d3a4c8..472e6ebfbd4d939bddb41771b5938c2b7595d6ff 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 d49227483bd3ae071242b0b2ff04236154080f68..d4961adb2a36df578388d1575fa1f6f205fed55d 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 ffb1958c015b645df78b7bdd80c83cb6415665ed..f7e41a338e82b88f01ba1c238082ad686f9f83ef 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