File size: 1,222 Bytes
a9b946e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
#!/usr/bin/env bash
export CUDA_VISIBLE_DEVICES="7"
. path.sh
set -ex
# epoch=1
# iter=2
mode=1
use_averaged_model=1
epoch=39
iter=116000
avg_list="11 12 13 14 15"
if [ $mode -eq 1 ]; then
for avg in $avg_list; do
./pruned_transducer_stateless5/decode.py \
--epoch $epoch \
--avg $avg \
--exp-dir ./pruned_transducer_stateless5/exp-L \
--max-duration 600 \
--decoding-method fast_beam_search \
--max-sym-per-frame 1 \
--num-encoder-layers 18 \
--dim-feedforward 2048 \
--nhead 8 \
--encoder-dim 512 \
--decoder-dim 512 \
--joiner-dim 512 \
--use-averaged-model $use_averaged_model
done
elif [ $mode -eq 2 ]; then
for avg in $avg_list; do
./pruned_transducer_stateless5/decode.py \
--iter $iter \
--avg $avg \
--exp-dir ./pruned_transducer_stateless5/exp-L \
--max-duration 600 \
--decoding-method greedy_search \
--max-sym-per-frame 1 \
--num-encoder-layers 18 \
--dim-feedforward 2048 \
--nhead 8 \
--encoder-dim 512 \
--decoder-dim 512 \
--joiner-dim 512 \
--use-averaged-model $use_averaged_model
done
else
echo "Unsupported mode $mode"
fi
|