csukuangfj commited on
Commit
3dbdbe1
1 Parent(s): b3c4165

Add pretrained model for the giga branch

Browse files
exp/giga/decode.sh ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ export CUDA_VISIBLE_DEVICES="7"
4
+
5
+ . path.sh
6
+
7
+ # for m in greedy_search fast_beam_search modified_beam_search; do
8
+ for m in greedy_search ; do
9
+ for iter in 468000; do
10
+ for avg in 16; do
11
+ ./lstm_transducer_stateless2/decode.py \
12
+ --use-giga-branch 1 \
13
+ --iter $iter \
14
+ --avg $avg \
15
+ --exp-dir lstm_transducer_stateless2/exp \
16
+ --max-duration 600 \
17
+ --num-encoder-layers 12 \
18
+ --rnn-hidden-size 1024 \
19
+ --decoding-method $m \
20
+ --use-averaged-model True \
21
+ --beam 4 \
22
+ --max-contexts 4 \
23
+ --max-states 8 \
24
+ --beam-size 4
25
+ done
26
+ done
27
+ done
exp/giga/decoder_jit_trace-iter-468000-avg-16-giga.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a37d7c11a48fce68eb34dfe4a0c9fa09116b7baa4755073d606c19cd02d9ecf5
3
+ size 1033475
exp/giga/encoder_jit_trace-iter-468000-avg-16-giga.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:673dc98d44530ca11edc6cb37acd8c385714a17fb908a78f69606e1a1af2291e
3
+ size 332784613
exp/giga/export.sh ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ export CUDA_VISIBLE_DEVICES=""
4
+
5
+ . path.sh
6
+
7
+ iter=468000
8
+ avg=16
9
+
10
+ iter=472000
11
+ avg=18
12
+
13
+ set -ex
14
+
15
+ if true; then
16
+ ./lstm_transducer_stateless2/export.py \
17
+ --use-giga-branch 1 \
18
+ --exp-dir ./lstm_transducer_stateless2/exp \
19
+ --bpe-model data/lang_bpe_500/bpe.model \
20
+ --iter $iter \
21
+ --avg $avg \
22
+ --pnnx 1
23
+ mv -v lstm_transducer_stateless2/exp/encoder_jit_trace-pnnx-giga.pt lstm_transducer_stateless2/exp/encoder_jit_trace-iter-$iter-avg-$avg-pnnx-giga.pt
24
+ mv -v lstm_transducer_stateless2/exp/decoder_jit_trace-pnnx-giga.pt lstm_transducer_stateless2/exp/decoder_jit_trace-iter-$iter-avg-$avg-pnnx-giga.pt
25
+ mv -v lstm_transducer_stateless2/exp/joiner_jit_trace-pnnx-giga.pt lstm_transducer_stateless2/exp/joiner_jit_trace-iter-$iter-avg-$avg-pnnx-giga.pt
26
+ fi
27
+
28
+ if true; then
29
+ ./lstm_transducer_stateless2/export.py \
30
+ --use-giga-branch 1 \
31
+ --exp-dir ./lstm_transducer_stateless2/exp \
32
+ --bpe-model data/lang_bpe_500/bpe.model \
33
+ --iter $iter \
34
+ --avg $avg \
35
+ --jit-trace 1
36
+ mv lstm_transducer_stateless2/exp/encoder_jit_trace-giga.pt lstm_transducer_stateless2/exp/encoder_jit_trace-iter-$iter-avg-$avg-giga.pt
37
+ mv lstm_transducer_stateless2/exp/decoder_jit_trace-giga.pt lstm_transducer_stateless2/exp/decoder_jit_trace-iter-$iter-avg-$avg-giga.pt
38
+ mv lstm_transducer_stateless2/exp/joiner_jit_trace-giga.pt lstm_transducer_stateless2/exp/joiner_jit_trace-iter-$iter-avg-$avg-giga.pt
39
+ fi
40
+
41
+ if true; then
42
+ ./lstm_transducer_stateless2/export.py \
43
+ --use-giga-branch 1 \
44
+ --exp-dir ./lstm_transducer_stateless2/exp \
45
+ --bpe-model data/lang_bpe_500/bpe.model \
46
+ --iter $iter \
47
+ --avg $avg \
48
+ --jit-trace 0
49
+ mv lstm_transducer_stateless2/exp/pretrained-giga.pt lstm_transducer_stateless2/exp/pretrained-iter-$iter-avg-$avg-giga.pt
50
+ fi
exp/giga/joiner_jit_trace-iter-468000-avg-16-giga.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55af03bb72da9be1255baab8cfa1b8218ae6c87071a437a4c213f143d649d9aa
3
+ size 3134718
exp/giga/pretrained-iter-468000-avg-16-giga.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac881ea4cad22925f461cdbf5cb0c8c7e3b5cbfb82b643f60c523013c0a02ab4
3
+ size 338850087