Commit
·
b2cbfae
1
Parent(s):
5fdafa8
Upload 3 files (#2)
Browse files- Upload 3 files (3df8e3163685ac80166adfdf2d34bb8966ee4e93)
Co-authored-by: fangjun <csukuangfj@users.noreply.huggingface.co>
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
|