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
|