export CUDA_VISIBLE_DEVICES=0

./zipformer/pretrained.py \
  --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 \
  --checkpoint ./zipformer/exp-small/pretrained.pt \
  --bpe-model ./data/lang_bpe_500/bpe.model \
  --method greedy_search \
  test_wavs/1089-134686-0001.wav \
  test_wavs/1221-135766-0001.wav \
  test_wavs/1221-135766-0002.wav