|
#!/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 |
|
|
