Commit ·
3df8e31
1
Parent(s): 5fdafa8
Upload 3 files
Browse filesSee https://github.com/k2-fsa/sherpa-onnx/pull/125
exp/export-onnx.sh
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env bash
|
| 2 |
+
|
| 3 |
+
# We use the model from
|
| 4 |
+
# https://huggingface.co/ezerhouni/icefall-librispeech-rnn-lm/tree/main
|
| 5 |
+
# as an example
|
| 6 |
+
|
| 7 |
+
export CUDA_VISIBLE_DEVICES=
|
| 8 |
+
|
| 9 |
+
if [ ! -f ./icefall-librispeech-rnn-lm/exp/pretrained.pt ]; then
|
| 10 |
+
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/ezerhouni/icefall-librispeech-rnn-lm
|
| 11 |
+
pushd icefall-librispeech-rnn-lm/exp
|
| 12 |
+
git lfs pull --include "pretrained.pt"
|
| 13 |
+
ln -s pretrained.pt epoch-99.pt
|
| 14 |
+
popd
|
| 15 |
+
fi
|
| 16 |
+
|
| 17 |
+
python3 ./export-onnx.py \
|
| 18 |
+
--exp-dir ./icefall-librispeech-rnn-lm/exp \
|
| 19 |
+
--epoch 99 \
|
| 20 |
+
--avg 1 \
|
| 21 |
+
--vocab-size 500 \
|
| 22 |
+
--embedding-dim 2048 \
|
| 23 |
+
--hidden-dim 2048 \
|
| 24 |
+
--num-layers 3 \
|
| 25 |
+
--tie-weights 1
|
| 26 |
+
|
exp/no-state-epoch-99-avg-1.int8.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9ebda265406cf60ba5448f18a8e376ebf39c11bfa22176fb599fcf7177ffff5d
|
| 3 |
+
size 102923418
|
exp/no-state-epoch-99-avg-1.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2b1327ce62db36d20c3c2a183226e48002140b894543c87a27f90803b94566d9
|
| 3 |
+
size 411052576
|