export Zipformer.streaming_forward for torch.jit.script (#4)

- export Zipformer.streaming_forward for torch.jit.script (cae03e042cdd76f1486f9dc4dae27e246fb0c638)


Co-authored-by: fangjun <[email protected]>

exp/cpu_jit.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aef1c155e3c8726546aa6fd285a24326666b95f89caed51a5747e518ff49174a
-size 379184446
+oid sha256:7e72d9330b56b09a93d983ee7a3118ba919fc71696ff360a48858d4757e5885f
+size 379191678

exp/export-stateless7-streaming-zh.sh

@@ -0,0 +1,23 @@
+#!/usr/bin/env bash
+
+. path.sh
+
+./pruned_transducer_stateless7_streaming/export-zh.py \
+  --lang-dir ./k2fsa-zipformer-chinese-english-mixed/data/lang_char_bpe \
+  --use-averaged-model 0 \
+  --epoch 99 \
+  --avg 1 \
+  --exp-dir ./k2fsa-zipformer-chinese-english-mixed/exp/ \
+  --jit 1 \
+  \
+  --decode-chunk-len 32 \
+  --num-encoder-layers "2,4,3,2,4" \
+  --feedforward-dims "1024,1024,1536,1536,1024" \
+  --nhead "8,8,8,8,8" \
+  --encoder-dims "384,384,384,384,384" \
+  --attention-dims "192,192,192,192,192" \
+  --encoder-unmasked-dims "256,256,256,256,256" \
+  --zipformer-downsampling-factors "1,2,4,8,2" \
+  --cnn-module-kernels "31,31,31,31,31" \
+  --decoder-dim 512 \
+  --joiner-dim 512