export-onnx-stateless2.sh

#!/usr/bin/env bash

export CUDA_VISIBLE_DEVICES=""

repo=./icefall_asr_wenetspeech_pruned_transducer_stateless2

if [ ! -f $repo/exp/epoch-99.pt ]; then
  pushd $repo/exp
  ln -s pretrained_epoch_10_avg_2.pt epoch-99.pt
  popd
fi

./pruned_transducer_stateless2/export-onnx.py \
  --lang-dir $repo/data/lang_char \
  --epoch 99 \
  --avg 1 \
  --exp-dir $repo/exp