File size: 365 Bytes
735a88e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#!/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