| export CUDA_VISIBLE_DEVICES="0" | |
| for m in greedy_search modified_beam_search fast_beam_search; do | |
| ./zipformer/decode.py \ | |
| --epoch 40 \ | |
| --avg 13 \ | |
| --exp-dir zipformer/exp-small \ | |
| --max-duration 600 \ | |
| --causal 0 \ | |
| --decoding-method $m \ | |
| --num-encoder-layers 2,2,2,2,2,2 \ | |
| --feedforward-dim 512,768,768,768,768,768 \ | |
| --encoder-dim 192,256,256,256,256,256 \ | |
| --encoder-unmasked-dim 192,192,192,192,192,192 | |