yfyeung commited on
Commit
30d17c8
1 Parent(s): 0c8bbdf

Add export for onnx

Browse files
exp/decoder-epoch-30-avg-5.int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d563d94327e19803e09f6ae76e924bd7a2df172b6be946aff365e6920ef094f
3
+ size 1307157
exp/decoder-epoch-30-avg-5.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:959040efc1d6644fb66b15c0191fd2af55fdecfb6806db997e3b311ce87fe5c3
3
+ size 2092272
exp/encoder-epoch-30-avg-5.int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b8fd2e98d257508ec12a4e24f719364d6d1b8ef3b2086592cd1f100c60f958e3
3
+ size 127114950
exp/encoder-epoch-30-avg-5.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca52724401e5130a9d0aff5d7f7422772b26bd8c999b7fbfd268985b784fddd3
3
+ size 293012698
exp/export-onnx.sh ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ export CUDA_VISIBLE_DEVICES=
4
+
5
+ ./pruned_transducer_stateless7/export-onnx.py \
6
+ --bpe-model ./data/en/lang_bpe_500/bpe.model \
7
+ --use-averaged-model 1 \
8
+ --epoch 30 \
9
+ --avg 5 \
10
+ --exp-dir ./pruned_transducer_stateless7/exp
exp/joiner-epoch-30-avg-5.int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f6f7a57a25ab99dcf22a295fd1b4d26dc81ec25f7580366887df5b98542ee754
3
+ size 259572
exp/joiner-epoch-30-avg-5.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ac4458e88682e905d0bc1612cfc705d6226c06fa51144bf3a1aedbcc6ea0b4c
3
+ size 1026462
exp/onnx_pretrained.sh ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+
4
+ ./pruned_transducer_stateless7/onnx_pretrained.py \
5
+ --encoder-model-filename ./pruned_transducer_stateless7/exp/encoder-epoch-9999-avg-1.onnx \
6
+ --decoder-model-filename ./pruned_transducer_stateless7/exp/decoder-epoch-9999-avg-1.onnx \
7
+ --joiner-model-filename ./pruned_transducer_stateless7/exp/joiner-epoch-9999-avg-1.onnx \
8
+ --tokens ./data/en/lang_bpe_500/tokens.txt \
9
+ ./test_wavs/1089-134686-0001.wav \
10
+ ./test_wavs/1221-135766-0001.wav \
11
+ ./test_wavs/1221-135766-0002.wav