set -e | |
export CUDA_VISIBLE_DEVICES=0 | |
. path.sh | |
epoch=38 | |
avg=10 | |
./pruned_transducer_stateless2/export.py \ | |
--exp-dir ./pruned_transducer_stateless2/exp \ | |
--bpe-model data/lang_bpe_500/bpe.model \ | |
--epoch $epoch \ | |
--avg $avg | |
mv -v ./pruned_transducer_stateless2/exp/pretrained.pt ./pruned_transducer_stateless2/exp/pretrained-epoch-${epoch}-avg-${avg}.pt | |