#!/usr/bin/env bash

set -e

export CUDA_VISIBLE_DEVICES=

epoch=99
avg=1

if [ ! -f pruned_transducer_stateless3/exp-0.9/epoch-99.pt ]; then
  mkdir -p pruned_transducer_stateless3/exp-0.9
  pushd pruned_transducer_stateless3/exp-0.9
  ln -s /ceph-fj/fangjun/open-source-2/icefall-models/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13/exp/pretrained-iter-1224000-avg-14.pt epoch-99.pt
  popd
fi

./pruned_transducer_stateless3/export.py \
  --exp-dir ./pruned_transducer_stateless3/exp-0.9 \
  --bpe-model data/lang_bpe_500/bpe.model \
  --epoch $epoch \
  --avg $avg \
  --jit 0 \
  --jit-trace 0 \
  --onnx 1