File size: 413 Bytes
2b86a06 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
#!/usr/bin/env bash
. path.sh
set -ex
for jit in 0 1; do
./pruned_transducer_stateless3/export.py \
--jit $jit \
--context-size 1 \
--exp-dir ./pruned_transducer_stateless3/exp-context-size-1 \
--epoch 29 \
--avg 5 \
--num-encoder-layers 12 \
--dim-feedforward 2048 \
--nhead 8 \
--encoder-dim 512 \
--context-size 1 \
--decoder-dim 512 \
--joiner-dim 512
done
|