csukuangfj
commited on
Commit
•
368f2f2
1
Parent(s):
1400551
add onnx models
Browse files- exp/onnx/decoder.onnx +3 -0
- exp/onnx/encoder.onnx +3 -0
- exp/onnx/export-3.sh +24 -0
- exp/onnx/joiner.onnx +3 -0
- exp/onnx/joiner_decoder_proj.onnx +3 -0
- exp/onnx/joiner_encoder_proj.onnx +3 -0
exp/onnx/decoder.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:762ecaa56fb83292ec4b6aba240a6b1b4532f0337a2e5bdd9100973478df7eef
|
3 |
+
size 1028608
|
exp/onnx/encoder.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7bc2d5fe5b5e39dc05464826f55515f2cd27fc287cbc742be549d763df74d050
|
3 |
+
size 329041122
|
exp/onnx/export-3.sh
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/usr/bin/env bash
|
2 |
+
|
3 |
+
set -e
|
4 |
+
|
5 |
+
export CUDA_VISIBLE_DEVICES=
|
6 |
+
|
7 |
+
epoch=99
|
8 |
+
avg=1
|
9 |
+
|
10 |
+
if [ ! -f pruned_transducer_stateless3/exp-0.9/epoch-99.pt ]; then
|
11 |
+
mkdir -p pruned_transducer_stateless3/exp-0.9
|
12 |
+
pushd pruned_transducer_stateless3/exp-0.9
|
13 |
+
ln -s /ceph-fj/fangjun/open-source-2/icefall-models/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13/exp/pretrained-iter-1224000-avg-14.pt epoch-99.pt
|
14 |
+
popd
|
15 |
+
fi
|
16 |
+
|
17 |
+
./pruned_transducer_stateless3/export.py \
|
18 |
+
--exp-dir ./pruned_transducer_stateless3/exp-0.9 \
|
19 |
+
--bpe-model data/lang_bpe_500/bpe.model \
|
20 |
+
--epoch $epoch \
|
21 |
+
--avg $avg \
|
22 |
+
--jit 0 \
|
23 |
+
--jit-trace 0 \
|
24 |
+
--onnx 1
|
exp/onnx/joiner.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:143b4ea96d4e2f230b7dcfafbceedab905b8222e05a4fcccf55bf41f5c1ef727
|
3 |
+
size 1026428
|
exp/onnx/joiner_decoder_proj.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:826a5570d1e18caa3d733211e7e52587be6841c7188f0bd6b0e2192b3a6564dc
|
3 |
+
size 1050898
|
exp/onnx/joiner_encoder_proj.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d2ad02e058ad80a52d8106bcc85ef78a693357a3d9a271fe5374f123020f5c1d
|
3 |
+
size 1050898
|