resum librispeech

examples/librispeech/asr1/local/test.sh

@@ -42,11 +42,6 @@ echo "chunk mode ${chunk_mode}"
 
 
 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
-    # format the reference test file
-    python utils/format_rsl.py \
-        --origin_ref data/manifest.test.raw \
-        --trans_ref data/manifest.test.text
-
     for type in attention; do
         echo "decoding ${type}"
         if [ ${chunk_mode} == true ];then
@@ -68,20 +63,11 @@ if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then
             echo "Failed in evaluation!"
             exit 1
         fi
-         # format the hyp file
-        python utils/format_rsl.py \
-            --origin_hyp ${ckpt_prefix}.${type}.rsl \
-            --trans_hyp ${ckpt_prefix}.${type}.rsl.text
-
-        python utils/compute-wer.py --char=1 --v=1 \
-            data/manifest.test.text ${ckpt_prefix}.${type}.rsl.text > ${ckpt_prefix}.${type}.error
-
         echo "decoding ${type} done."
     done
 fi
 
-if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
-    for type in ctc_greedy_search; do
+for type in ctc_greedy_search; do
     echo "decoding ${type}"
     if [ ${chunk_mode} == true ];then
         # stream decoding only support batchsize=1
@@ -102,21 +88,12 @@ if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
         echo "Failed in evaluation!"
         exit 1
     fi
+    echo "decoding ${type} done."
+done
 
-        # format the hyp file
-        python utils/format_rsl.py \
-            --origin_hyp ${ckpt_prefix}.${type}.rsl \
-            --trans_hyp ${ckpt_prefix}.${type}.rsl.text
-
-        python utils/compute-wer.py --char=1 --v=1 \
-            data/manifest.test.text ${ckpt_prefix}.${type}.rsl.text > ${ckpt_prefix}.${type}.error
 
-        echo "decoding ${type} done."
-    done
-fi
 
-if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
-    for type in ctc_prefix_beam_search attention_rescoring; do
+for type in ctc_prefix_beam_search attention_rescoring; do
     echo "decoding ${type}"
     batch_size=1
     python3 -u ${BIN_DIR}/test.py \
@@ -132,34 +109,9 @@ if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
         echo "Failed in evaluation!"
         exit 1
     fi
-        # format the hyp file
-        python utils/format_rsl.py \
-            --origin_hyp ${ckpt_prefix}.${type}.rsl \
-            --trans_hyp ${ckpt_prefix}.${type}.rsl.text
-
-        python utils/compute-wer.py --char=1 --v=1 \
-            data/manifest.test.text ${ckpt_prefix}.${type}.rsl.text > ${ckpt_prefix}.${type}.error
-
     echo "decoding ${type} done."
-    done
-fi
-echo "Finished"
-
-
-if [ ${stage} -le 101 ] && [ ${stop_stage} -ge 101 ]; then
-    # format the reference test file for sclite
-    python utils/format_rsl.py \
-        --origin_ref data/manifest.test.raw \
-        --trans_ref_sclite data/manifest.test.raw.text.sclite
+done
 
-    for type in attention ctc_greedy_search ctc_prefix_beam_search attention_rescoring; do
-        python utils/format_rsl.py \
-            --origin_hyp ${ckpt_prefix}.${type}.rsl
-            --trans_hyp_sclite ${ckpt_prefix}.${type}.rsl.text.sclite
-
-        mkdir -p ${ckpt_prefix}.${type}_sclite
-        sclite -i wsj -r data/manifest.test.raw.text.sclite -h  ${ckpt_prefix}.${type}.rsl.text.sclite  -e utf-8 -o all -O ${ckpt_prefix}.${type}_sclite
-    done
-fi
+echo "Finished"
 
 exit 0