mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-06 07:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

doc_str fixed

Browse Source
This commit is contained in:
jinzr 2024-01-16 15:29:44 +08:00
parent d90a24b946
commit c6669de38d
3 changed files with 44 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
egs/aishell/ASR/zipformer/decode_bbpe.py
View File

@ -21,30 +21,33 @@
""" """
Usage: Usage:
(1) greedy search (1) greedy search
./zipformer_bbpe/decode.py \ ./zipformer/decode_bbpe.py \
--epoch 35 \ --epoch 35 \
--avg 15 \ --avg 15 \
--exp-dir ./zipformer_bbpe/exp \ --exp-dir ./zipformer/exp_bbpe \
--lang-dir data/lang_char \ --lang-dir data/lang_bbpe_500 \
--bpe-model data/lang_bbpe_500/bbpe.model \
--max-duration 600 \ --max-duration 600 \
--decoding-method greedy_search --decoding-method greedy_search
(2) modified beam search (2) modified beam search
./zipformer_bbpe/decode.py \ ./zipformer/decode_bbpe.py \
--epoch 35 \ --epoch 35 \
--avg 15 \ --avg 15 \
--exp-dir ./zipformer_bbpe/exp \ --exp-dir ./zipformer/exp_bbpe \
--lang-dir data/lang_char \ --lang-dir data/lang_bbpe_500 \
--bpe-model data/lang_bbpe_500/bbpe.model \
--max-duration 600 \ --max-duration 600 \
--decoding-method modified_beam_search \ --decoding-method modified_beam_search \
--beam-size 4 --beam-size 4
(3) fast beam search (trivial_graph) (3) fast beam search (trivial_graph)
./zipformer_bbpe/decode.py \ ./zipformer/decode_bbpe.py \
--epoch 35 \ --epoch 35 \
--avg 15 \ --avg 15 \
--exp-dir ./zipformer_bbpe/exp \ --exp-dir ./zipformer/exp_bbpe \
--lang-dir data/lang_char \ --lang-dir data/lang_bbpe_500 \
--bpe-model data/lang_bbpe_500/bbpe.model \
--max-duration 600 \ --max-duration 600 \
--decoding-method fast_beam_search \ --decoding-method fast_beam_search \
--beam 20.0 \ --beam 20.0 \
@ -52,11 +55,12 @@ Usage:
--max-states 64 --max-states 64
(4) fast beam search (LG) (4) fast beam search (LG)
./zipformer_bbpe/decode.py \ ./zipformer/decode_bbpe.py \
--epoch 30 \ --epoch 30 \
--avg 15 \ --avg 15 \
--exp-dir ./zipformer_bbpe/exp \ --exp-dir ./zipformer/exp_bbpe \
--lang-dir data/lang_char \ --lang-dir data/lang_bbpe_500 \
--bpe-model data/lang_bbpe_500/bbpe.model \
--max-duration 600 \ --max-duration 600 \
--decoding-method fast_beam_search_LG \ --decoding-method fast_beam_search_LG \
--beam 20.0 \ --beam 20.0 \
@ -64,11 +68,12 @@ Usage:
--max-states 64 --max-states 64
(5) fast beam search (nbest oracle WER) (5) fast beam search (nbest oracle WER)
./zipformer_bbpe/decode.py \ ./zipformer/decode_bbpe.py \
--epoch 35 \ --epoch 35 \
--avg 15 \ --avg 15 \
--exp-dir ./zipformer_bbpe/exp \ --exp-dir ./zipformer/exp_bbpe \
--lang-dir data/lang_char \ --lang-dir data/lang_bbpe_500 \
--bpe-model data/lang_bbpe_500/bbpe.model \
--max-duration 600 \ --max-duration 600 \
--decoding-method fast_beam_search_nbest_oracle \ --decoding-method fast_beam_search_nbest_oracle \
--beam 20.0 \ --beam 20.0 \

40
egs/aishell/ASR/zipformer/pretrained_bbpe.py
View File

@ -25,18 +25,18 @@ dataset, you should change the argument values according to your dataset.
- For non-streaming model: - For non-streaming model:
./zipformer_bbpe/export.py \ ./zipformer/export.py \
--exp-dir ./zipformer_bbpe/exp \ --exp-dir ./zipformer/exp_bbpe \
--bpe ./data/lang_bbpe_500/bbpe.model \ --tokens ./data/lang_bbpe_500/tokens.txt \
--epoch 30 \ --epoch 30 \
--avg 9 --avg 9
- For streaming model: - For streaming model:
./zipformer_bbpe/export.py \ ./zipformer/export.py \
--exp-dir ./zipformer_bbpe/exp \ --exp-dir ./zipformer/exp_bbpe \
--causal 1 \ --causal 1 \
--bpe ./data/lang_bbpe_500/bbpe.model \ --tokens ./data/lang_bbpe_500/tokens.txt \
--epoch 30 \ --epoch 30 \
--avg 9 --avg 9
@ -45,24 +45,24 @@ Usage of this script:
- For non-streaming model: - For non-streaming model:
(1) greedy search (1) greedy search
./zipformer_bbpe/pretrained.py \ ./zipformer/pretrained_bbpe.py \
--checkpoint ./zipformer_bbpe/exp/pretrained.pt \ --checkpoint ./zipformer/exp_bbpe/pretrained.pt \
--bpe ./data/lang_bbpe_500/bbpe.model \ --bpe ./data/lang_bbpe_500/bbpe.model \
--method greedy_search \ --method greedy_search \
/path/to/foo.wav \ /path/to/foo.wav \
/path/to/bar.wav /path/to/bar.wav
(2) modified beam search (2) modified beam search
./zipformer_bbpe/pretrained.py \ ./zipformer/pretrained_bbpe.py \
--checkpoint ./zipformer_bbpe/exp/pretrained.pt \ --checkpoint ./zipformer/exp_bbpe/pretrained.pt \
--bpe ./data/lang_bbpe_500/bbpe.model \ --bpe ./data/lang_bbpe_500/bbpe.model \
--method modified_beam_search \ --method modified_beam_search \
/path/to/foo.wav \ /path/to/foo.wav \
/path/to/bar.wav /path/to/bar.wav
(3) fast beam search (3) fast beam search
./zipformer_bbpe/pretrained.py \ ./zipformer/pretrained_bbpe.py \
--checkpoint ./zipformer_bbpe/exp/pretrained.pt \ --checkpoint ./zipformer/exp_bbpe/pretrained.pt \
--bpe ./data/lang_bbpe_500/bbpe.model \ --bpe ./data/lang_bbpe_500/bbpe.model \
--method fast_beam_search \ --method fast_beam_search \
/path/to/foo.wav \ /path/to/foo.wav \
@ -71,8 +71,8 @@ Usage of this script:
- For streaming model: - For streaming model:
(1) greedy search (1) greedy search
./zipformer_bbpe/pretrained.py \ ./zipformer/pretrained_bbpe.py \
--checkpoint ./zipformer_bbpe/exp/pretrained.pt \ --checkpoint ./zipformer/exp_bbpe/pretrained.pt \
--causal 1 \ --causal 1 \
--chunk-size 16 \ --chunk-size 16 \
--left-context-frames 128 \ --left-context-frames 128 \
@ -82,8 +82,8 @@ Usage of this script:
/path/to/bar.wav /path/to/bar.wav
(2) modified beam search (2) modified beam search
./zipformer_bbpe/pretrained.py \ ./zipformer/pretrained_bbpe.py \
--checkpoint ./zipformer_bbpe/exp/pretrained.pt \ --checkpoint ./zipformer/exp_bbpe/pretrained.pt \
--causal 1 \ --causal 1 \
--chunk-size 16 \ --chunk-size 16 \
--left-context-frames 128 \ --left-context-frames 128 \
@ -93,8 +93,8 @@ Usage of this script:
/path/to/bar.wav /path/to/bar.wav
(3) fast beam search (3) fast beam search
./zipformer_bbpe/pretrained.py \ ./zipformer/pretrained_bbpe.py \
--checkpoint ./zipformer_bbpe/exp/pretrained.pt \ --checkpoint ./zipformer/exp_bbpe/pretrained.pt \
--causal 1 \ --causal 1 \
--chunk-size 16 \ --chunk-size 16 \
--left-context-frames 128 \ --left-context-frames 128 \
@ -104,9 +104,9 @@ Usage of this script:
/path/to/bar.wav /path/to/bar.wav
You can also use `./zipformer_bbpe/exp/epoch-xx.pt`. You can also use `./zipformer/exp_bbpe/epoch-xx.pt`.
Note: ./zipformer_bbpe/exp/pretrained.pt is generated by ./zipformer_bbpe/export.py Note: ./zipformer/exp_bbpe/pretrained.pt is generated by ./zipformer/export_bbpe.py
""" """

12
egs/aishell/ASR/zipformer/train_bbpe.py
View File

@ -24,25 +24,21 @@ Usage:
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
./zipformer/train.py \ ./zipformer/train_bbpe.py \
--world-size 8 \ --world-size 8 \
--num-epochs 12 \ --num-epochs 12 \
--start-epoch 1 \ --start-epoch 1 \
--exp-dir zipformer/exp \ --exp-dir zipformer/exp_bbpe \
--training-subset L
--lr-epochs 1.5 \
--max-duration 350 --max-duration 350
# For mix precision training: # For mix precision training:
./zipformer/train.py \ ./zipformer/train_bbpe.py \
--world-size 8 \ --world-size 8 \
--num-epochs 12 \ --num-epochs 12 \
--start-epoch 1 \ --start-epoch 1 \
--use-fp16 1 \ --use-fp16 1 \
--exp-dir zipformer/exp \ --exp-dir zipformer/exp_bbpe \
--training-subset L \
--lr-epochs 1.5 \
--max-duration 750 --max-duration 750
""" """