ezerhouni csukuangfj commited on
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