|
#!/usr/bin/env bash |
|
|
|
. path.sh |
|
|
|
export CUDA_VISIBLE_DEVICES="" |
|
|
|
set -ex |
|
|
|
epoch=15 |
|
avg=3 |
|
jit=1 |
|
|
|
dir=./conv_emformer_transducer_stateless2 |
|
|
|
./conv_emformer_transducer_stateless2/export.py \ |
|
--exp-dir $dir/exp-L-small \ |
|
--lang-dir ./data/lang_char \ |
|
--epoch $epoch \ |
|
--avg $avg \ |
|
--use-averaged-model 1 \ |
|
\ |
|
--encoder-dim 256 \ |
|
--nhead 4 \ |
|
--dim-feedforward 1024 \ |
|
--num-encoder-layers 5 \ |
|
--chunk-length 32 \ |
|
--cnn-module-kernel 31 \ |
|
--left-context-length 32 \ |
|
--right-context-length 8 \ |
|
--memory-size 32 \ |
|
--jit $jit |
|
|
|
if [ $jit -eq 0 ]; then |
|
pushd $dir/exp-L-small |
|
mv pretrained.pt pretrained-epoch-$epoch-avg-$avg.pt |
|
popd |
|
fi |
|
|
|
if [ $jit -eq 1 ]; then |
|
pushd $dir/exp-L-small |
|
mv cpu_jit.pt cpu_jit-epoch-$epoch-avg-$avg.pt |
|
popd |
|
fi |
|
|
