csukuangfj commited on
Commit
2b86a06
1 Parent(s): 9ed5963

add script files.

Browse files
data/lang_char/Linv.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03fab81938e46d45bd23ce89d322f53ae038f7e67eec16b9ea2a9d57d4ba3b99
3
+ size 6721639
exp/decode-pruned-transducer-stateless3-2.sh ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ . path.sh
4
+
5
+ export CUDA_VISIBLE_DEVICES=7
6
+
7
+ set -ex
8
+
9
+ for epoch in 39 38 37 36 35 34 33 32 32; do
10
+ for avg in 3 4 5 6 7 8 9 10 11; do
11
+ for m in greedy_search; do
12
+ ./pruned_transducer_stateless3/decode.py \
13
+ --exp-dir ./pruned_transducer_stateless3/exp-context-size-1 \
14
+ --epoch $epoch \
15
+ --avg $avg \
16
+ --use-averaged-model 1 \
17
+ --max-duration 600 \
18
+ --decoding-method $m \
19
+ --num-encoder-layers 12 \
20
+ --dim-feedforward 2048 \
21
+ --nhead 8 \
22
+ --context-size 1 \
23
+ --encoder-dim 512 \
24
+ --decoder-dim 512 \
25
+ --joiner-dim 512
26
+ done
27
+ done
28
+ done
exp/export.sh ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ . path.sh
3
+
4
+ set -ex
5
+
6
+ for jit in 0 1; do
7
+ ./pruned_transducer_stateless3/export.py \
8
+ --jit $jit \
9
+ --context-size 1 \
10
+ --exp-dir ./pruned_transducer_stateless3/exp-context-size-1 \
11
+ --epoch 29 \
12
+ --avg 5 \
13
+ --num-encoder-layers 12 \
14
+ --dim-feedforward 2048 \
15
+ --nhead 8 \
16
+ --encoder-dim 512 \
17
+ --context-size 1 \
18
+ --decoder-dim 512 \
19
+ --joiner-dim 512
20
+ done
exp/pretrained.sh ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ . path.sh
4
+
5
+ for m in greedy_search modified_beam_search fast_beam_search beam_search; do
6
+ ./pruned_transducer_stateless3/pretrained.py \
7
+ --checkpoint ./icefall-aishell-pruned-transducer-stateless3-2022-06-20/exp/pretrained-epoch-29-avg-5-torch-1.10.pt \
8
+ --lang-dir ./icefall-aishell-pruned-transducer-stateless3-2022-06-20/data/lang_char \
9
+ --method $m \
10
+ ./icefall-aishell-pruned-transducer-stateless3-2022-06-20/test_wavs/BAC009S0764W0121.wav \
11
+ ./icefall-aishell-pruned-transducer-stateless3-2022-06-20/test_wavs/BAC009S0764W0122.wav \
12
+ ./icefall-aishell-pruned-transducer-stateless3-2022-06-20/test_wavs/BAC009S0764W0123.wav
13
+ done
exp/run-pruned-transducer-stateless3-2.sh ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ . path.sh
4
+
5
+ # export CUDA_VISIBLE_DEVICES="4,5,6,7"
6
+ export CUDA_VISIBLE_DEVICES="0,1,2,3"
7
+
8
+ ./pruned_transducer_stateless3/train.py \
9
+ --exp-dir ./pruned_transducer_stateless3/exp-context-size-1 \
10
+ --world-size 4 \
11
+ --max-duration 200 \
12
+ --datatang-prob 0.5 \
13
+ --start-epoch 31 \
14
+ --num-epochs 40 \
15
+ --use-fp16 1 \
16
+ --num-encoder-layers 12 \
17
+ --dim-feedforward 2048 \
18
+ --nhead 8 \
19
+ --encoder-dim 512 \
20
+ --context-size 1 \
21
+ --decoder-dim 512 \
22
+ --joiner-dim 512 \
23
+ --master-port 12356