refactor: pass decoding

wenet-e2e · Dec 12, 2023 · b8cb316 · b8cb316
1 parent 7d4dede
commit b8cb316
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/examples/aishell/NST/run_nst.sh b/examples/aishell/NST/run_nst.sh
@@ -209,7 +209,6 @@ if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
       --beam_size 10 \
       --batch_size 1 \
       --penalty 0.0 \
-      --dict $dict \
       --ctc_weight $ctc_weight \
       --reverse_weight $reverse_weight \
       --result_file $dev_dir/text \
@@ -268,7 +267,6 @@ if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
     --beam_size 10 \
     --batch_size 1 \
     --penalty 0.0 \
-    --dict $dict \
     --ctc_weight $ctc_weight \
     --reverse_weight $reverse_weight \
     --result_file data/train/${dir_split}data_sublist${job_num}/${hypo_name} \

diff --git a/examples/aishell/rnnt/run.sh b/examples/aishell/rnnt/run.sh
@@ -94,7 +94,7 @@ if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
   echo "<sos/eos> 2" >> $dict
   tools/text2token.py -s 1 -n 1 data/train/text | cut -f 2- -d" " \
     | tr " " "\n" | sort | uniq | grep -a -v -e '^\s*$' | \
-    awk '{print $0 " " NR+1}' >> ${dict}
+    awk '{print $0 " " NR+2}' >> ${dict}
 fi
 
 if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then