Views
No views yet
1./pruned_transducer_stateless5/train.py \
2 --exp-dir pruned_transducer_stateless5/exp \
3 --num-encoder-layers 18 \
4 --dim-feedforward 2048 \
5 --nhead 8 \
6 --encoder-dim 512 \
7 --decoder-dim 512 \
8 --joiner-dim 512 \
9 --full-libri 1 \
10 --dynamic-chunk-training 1 \
11 --causal-convolution 1 \
12 --short-chunk-size 20 \
13 --num-left-chunks 4 \
14 --max-duration 300 \
15 --world-size 4 \
16 --start-epoch 1 \
17 --num-epochs 251decoding_method="greedy_search" # "fast_beam_search", "modified_beam_search"
2for chunk in 2 4 8 16; do
3 for left in 32 64; do
4 ./pruned_transducer_stateless5/decode.py \
5 --num-encoder-layers 18 \
6 --dim-feedforward 2048 \
7 --nhead 8 \
8 --encoder-dim 512 \
9 --decoder-dim 512 \
10 --joiner-dim 512 \
11 --simulate-streaming 1 \
12 --decode-chunk-size ${chunk} \
13 --left-context ${left} \
14 --causal-convolution 1 \
15 --epoch 25 \
16 --avg 5 \
17 --exp-dir ./pruned_transducer_stateless5/exp \
18 --max-sym-per-frame 1 \
19 --max-duration 1000 \
20 --decoding-method ${decoding_method}
21 done
22done1./pruned_transducer_stateless5/export.py \
2 --streaming-model 1 \
3 --causal-convolution 1 \
4 --num-encoder-layers 18 \
5 --dim-feedforward 2048 \
6 --nhead 8 \
7 --encoder-dim 512 \
8 --decoder-dim 512 \
9 --joiner-dim 512 \
10 --epoch 25 \
11 --avg 5 \
12 --exp-dir ./pruned_transducer_stateless5/exp
13
14./pruned_transducer_stateless5/export.py \
15 --streaming-model 1 \
16 --causal-convolution 1 \
17 --num-encoder-layers 18 \
18 --dim-feedforward 2048 \
19 --nhead 8 \
20 --encoder-dim 512 \
21 --decoder-dim 512 \
22 --joiner-dim 512 \
23 --epoch 25 \
24 --avg 5 \
25 --exp-dir ./pruned_transducer_stateless5/exp \
26 --jit 1