re-org the bbpe recipe for aishell

k2-fsa · Jan 16, 2024 · 674390e · 674390e
1 parent ad94191
commit 674390e
Show file tree

Hide file tree

Showing 21 changed files with 17 additions and 439 deletions.
diff --git a/egs/aishell/ASR/RESULTS.md b/egs/aishell/ASR/RESULTS.md
@@ -4,7 +4,7 @@
 
 #### Zipformer (Byte-level BPE)
 
-[./zipformer_bbpe](./zipformer_bbpe/)
+[./zipformer](./zipformer/)
 
 It's reworked Zipformer with Pruned RNNT loss, trained with Byte-level BPE, `vocab_size` set to 500.
 
@@ -21,14 +21,14 @@ It's reworked Zipformer with Pruned RNNT loss, trained with Byte-level BPE, `voc
 
 export CUDA_VISIBLE_DEVICES="0,1"
 
-./zipformer_bbpe/train.py \
+./zipformer/train_bbpe.py \
   --world-size 2 \
   --num-epochs 40 \
   --start-epoch 1 \
   --use-fp16 1 \
   --context-size 2 \
   --enable-musan 0 \
-  --exp-dir zipformer/exp \
+  --exp-dir zipformer/exp_bbpe \
   --max-duration 1000 \
   --enable-musan 0 \
   --base-lr 0.045 \
@@ -40,11 +40,11 @@ export CUDA_VISIBLE_DEVICES="0,1"
 Command for decoding is:
 ```bash
 for m in greedy_search modified_beam_search fast_beam_search ; do
-  ./zipformer/decode.py \
+  ./zipformer/decode_bbpe.py \
     --epoch 40 \
     --avg 10 \
     --exp-dir ./zipformer_bbpe/exp \
-    --lang-dir data/lang_bbpe_500 \
+    --bpe-model data/lang_bbpe_500/bbpe.model \
     --context-size 2 \
     --decoding-method $m
 done

diff --git a/egs/aishell/ASR/zipformer_bbpe/decode.py → egs/aishell/ASR/zipformer/decode_bbpe.py b/egs/aishell/ASR/zipformer_bbpe/decode.py → egs/aishell/ASR/zipformer/decode_bbpe.py
@@ -93,7 +93,6 @@
 from asr_datamodule import AishellAsrDataModule
 from beam_search import (
     beam_search,
-    fast_beam_search_nbest,
     fast_beam_search_nbest_oracle,
     fast_beam_search_one_best,
     greedy_search,

diff --git a/...R/zipformer_bbpe/export-onnx-streaming.py → ...R/zipformer/export-onnx-streaming_bbpe.py b/...R/zipformer_bbpe/export-onnx-streaming.py → ...R/zipformer/export-onnx-streaming_bbpe.py
diff --git a/...aishell/ASR/zipformer_bbpe/export-onnx.py → ...aishell/ASR/zipformer/export-onnx_bbpe.py b/...aishell/ASR/zipformer_bbpe/export-onnx.py → ...aishell/ASR/zipformer/export-onnx_bbpe.py
diff --git a/egs/aishell/ASR/zipformer_bbpe/export.py → egs/aishell/ASR/zipformer/export_bbpe.py b/egs/aishell/ASR/zipformer_bbpe/export.py → egs/aishell/ASR/zipformer/export_bbpe.py
diff --git a/...hell/ASR/zipformer_bbpe/jit_pretrained.py → ...hell/ASR/zipformer/jit_pretrained_bbpe.py b/...hell/ASR/zipformer_bbpe/jit_pretrained.py → ...hell/ASR/zipformer/jit_pretrained_bbpe.py
diff --git a/egs/aishell/ASR/zipformer_bbpe/onnx_check.py → egs/aishell/ASR/zipformer/onnx_check_bbpe.py b/egs/aishell/ASR/zipformer_bbpe/onnx_check.py → egs/aishell/ASR/zipformer/onnx_check_bbpe.py
diff --git a/egs/aishell/ASR/zipformer_bbpe/pretrained.py → egs/aishell/ASR/zipformer/pretrained_bbpe.py b/egs/aishell/ASR/zipformer_bbpe/pretrained.py → egs/aishell/ASR/zipformer/pretrained_bbpe.py