csukuangfj's picture
Add training and decoding scripts.
97c5de0
raw
history blame
516 Bytes
#!/usr/bin/env bash
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