./zipformer/export.py \
  --exp-dir ./zipformer/exp-small \
  --bpe-model data/lang_bpe_500/bpe.model \
  --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 \
  --epoch 50 \
  --avg 23 \
  --jit 1