export Zipformer.streaming_forward for torch.jit.script
#4
by
csukuangfj
- opened
- exp/cpu_jit.pt +2 -2
- exp/export-stateless7-streaming-zh.sh +23 -0
exp/cpu_jit.pt
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7e72d9330b56b09a93d983ee7a3118ba919fc71696ff360a48858d4757e5885f
|
3 |
+
size 379191678
|
exp/export-stateless7-streaming-zh.sh
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/usr/bin/env bash
|
2 |
+
|
3 |
+
. path.sh
|
4 |
+
|
5 |
+
./pruned_transducer_stateless7_streaming/export-zh.py \
|
6 |
+
--lang-dir ./k2fsa-zipformer-chinese-english-mixed/data/lang_char_bpe \
|
7 |
+
--use-averaged-model 0 \
|
8 |
+
--epoch 99 \
|
9 |
+
--avg 1 \
|
10 |
+
--exp-dir ./k2fsa-zipformer-chinese-english-mixed/exp/ \
|
11 |
+
--jit 1 \
|
12 |
+
\
|
13 |
+
--decode-chunk-len 32 \
|
14 |
+
--num-encoder-layers "2,4,3,2,4" \
|
15 |
+
--feedforward-dims "1024,1024,1536,1536,1024" \
|
16 |
+
--nhead "8,8,8,8,8" \
|
17 |
+
--encoder-dims "384,384,384,384,384" \
|
18 |
+
--attention-dims "192,192,192,192,192" \
|
19 |
+
--encoder-unmasked-dims "256,256,256,256,256" \
|
20 |
+
--zipformer-downsampling-factors "1,2,4,8,2" \
|
21 |
+
--cnn-module-kernels "31,31,31,31,31" \
|
22 |
+
--decoder-dim 512 \
|
23 |
+
--joiner-dim 512
|