#!/usr/bin/env bash

export CUDA_VISIBLE_DEVICES=""
set -ex

dir=./k2fsa-zipformer-bilingual-zh-en-t
if [ ! -f $dir/exp/epoch-99.pt ]; then
  pushd $dir/exp
  ln -s pretrained.pt epoch-99.pt
  popd
fi

./pruned_transducer_stateless7_streaming/export-onnx-zh.py \
  --tokens $dir/data/lang_char_bpe/tokens.txt \
  --use-averaged-model 0 \
  --epoch 99 \
  --avg 1 \
  --exp-dir $dir/exp/ \
  --decode-chunk-len 96 \
  \
  --num-encoder-layers 2,2,2,2,2 \
  --feedforward-dims 768,768,768,768,768 \
  --nhead 4,4,4,4,4 \
  --encoder-dims 256,256,256,256,256 \
  --attention-dims 192,192,192,192,192 \
  --encoder-unmasked-dims 192,192,192,192,192 \
  \
  --zipformer-downsampling-factors "1,2,4,8,2" \
  --cnn-module-kernels "31,31,31,31,31" \
  --decoder-dim 512 \
  --joiner-dim 512