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
|