csukuangfj's picture
Add export script.
57bdcfe
#!/usr/bin/env bash
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