csukuangfj commited on
Commit
cae03e0
1 Parent(s): 706c9b7

export Zipformer.streaming_forward for torch.jit.script

Browse files
exp/cpu_jit.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:aef1c155e3c8726546aa6fd285a24326666b95f89caed51a5747e518ff49174a
3
- size 379184446
 
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