|
#!/usr/bin/env bash |
|
|
|
set -e |
|
|
|
export CUDA_VISIBLE_DEVICES=0 |
|
|
|
. path.sh |
|
epoch=39 |
|
avg=13 |
|
use_averaged_model=true |
|
|
|
|
|
./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 \ |
|
--use-averaged-model $use_averaged_model |
|
|
|
mv -v ./pruned_transducer_stateless5/exp-L/pretrained.pt ./pruned_transducer_stateless5/exp-L/pretrained-epoch-${epoch}-avg-${avg}-use-averaged-model-${use_averaged_model}.pt |
|
|
|
|