csukuangfj's picture
add onnx models
368f2f2
raw
history blame
629 Bytes
#!/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