export CUDA_VISIBLE_DEVICES=0 | |
set -ex | |
epoch=39 | |
avg=7 | |
./pruned_transducer_stateless5/export.py \ | |
--exp-dir ./pruned_transducer_stateless5/exp-L \ | |
--bpe-model data/lang_bpe_500/bpe.model \ | |
--epoch $epoch \ | |
--avg $avg \ | |
--num-encoder-layers 18 \ | |
--dim-feedforward 2048 \ | |
--nhead 8 \ | |
--encoder-dim 512 \ | |
--decoder-dim 512 \ | |
--joiner-dim 512 | |
mv -v ./pruned_transducer_stateless5/exp-L/pretrained.pt ./pruned_transducer_stateless5/exp-L/pretrained-epoch-${epoch}-avg-${avg}.pt | |