csukuangfj commited on
Commit
3a3a0e7
1 Parent(s): e7e7424

add scripts.

Browse files
exp/decode-pruned-rnnt-3-giga.sh ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ export CUDA_VISIBLE_DEVICES="7"
4
+
5
+ set -ex
6
+
7
+ for iter in 1224000; do
8
+ for avg in 14; do
9
+ ./pruned_transducer_stateless3/decode-giga.py \
10
+ --iter $iter \
11
+ --avg $avg \
12
+ --exp-dir ./pruned_transducer_stateless3/exp-0.9 \
13
+ --max-duration 600 \
14
+ --decoding-method fast_beam_search \
15
+ --max-sym-per-frame 1
16
+ done
17
+ done
exp/decode-pruned-rnnt-3.sh ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ export CUDA_VISIBLE_DEVICES="7"
4
+
5
+ set -ex
6
+
7
+ for iter in 1224000; do
8
+ for avg in 14; do
9
+ for method in greedy_search modified_beam_search fast_beam_search ; do
10
+ ./pruned_transducer_stateless3/decode.py \
11
+ --iter $iter \
12
+ --avg $avg \
13
+ --exp-dir ./pruned_transducer_stateless3/exp-0.9/ \
14
+ --max-duration 600 \
15
+ --decoding-method $method \
16
+ --max-sym-per-frame 1 \
17
+ --beam 4 \
18
+ --max-contexts 32
19
+ done
20
+ done
21
+ done
exp/export.sh ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ set -e
4
+
5
+ export CUDA_VISIBLE_DEVICES=7
6
+
7
+ iter=1224000
8
+ avg=14
9
+
10
+
11
+ ./pruned_transducer_stateless3/export.py \
12
+ --exp-dir ./pruned_transducer_stateless3/exp-0.9 \
13
+ --bpe-model data/lang_bpe_500/bpe.model \
14
+ --iter $iter \
15
+ --avg $avg
16
+
17
+ mv -v ./pruned_transducer_stateless3/exp-0.9/pretrained.pt ./pruned_transducer_stateless3/exp-0.9/pretrained-iter-${iter}-avg-${avg}.pt
exp/run-pruned-rnnt-3-2.sh ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
4
+
5
+ ./pruned_transducer_stateless3/train.py \
6
+ --world-size 8 \
7
+ --num-epochs 30 \
8
+ --start-epoch 0 \
9
+ --full-libri 1 \
10
+ --exp-dir pruned_transducer_stateless3/exp-0.9 \
11
+ --max-duration 300 \
12
+ --use-fp16 1 \
13
+ --lr-epochs 4 \
14
+ --num-workers 2 \
15
+ --giga-prob 0.9