csukuangfj
commited on
Commit
•
c6c9733
1
Parent(s):
f5c7845
add more models
Browse files- exp/cpu_jit-epoch-15-avg-3.pt +3 -0
- exp/epoch-12.pt +3 -0
- exp/epoch-13.pt +3 -0
- exp/epoch-14.pt +3 -0
- exp/epoch-15.pt +3 -0
- exp/pretrained-epoch-15-avg-3.pt +3 -0
- export.sh +43 -0
exp/cpu_jit-epoch-15-avg-3.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5ff482b96861d7f1cca41073b2aac801abb28f164572ffc067c6d639891fb39a
|
3 |
+
size 74493100
|
exp/epoch-12.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0fa3d31b6198b5fdab64a8a0b7e4f7d2747b9766404406d0283563062c2fea41
|
3 |
+
size 297865749
|
exp/epoch-13.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:03244df33e7257bc233a68663de128005ed1103b20eca2bd09cb55f8e5ef8ced
|
3 |
+
size 297865813
|
exp/epoch-14.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:715511e2e581161b6183d3aa16dbaa0cf484a73d6f4d0246843586367781c4ce
|
3 |
+
size 297865877
|
exp/epoch-15.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8fc0cd9457bda86d7dc53cd403c78e7c99865ce7cd76746ac33a35b3722f1d74
|
3 |
+
size 297865941
|
exp/pretrained-epoch-15-avg-3.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7b6433ce05aa002868da12339146f015e8ddf3f44d3946eec232604e5d1b8a91
|
3 |
+
size 74474581
|
export.sh
ADDED
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/usr/bin/env bash
|
2 |
+
|
3 |
+
. path.sh
|
4 |
+
|
5 |
+
export CUDA_VISIBLE_DEVICES=""
|
6 |
+
|
7 |
+
set -ex
|
8 |
+
|
9 |
+
epoch=15
|
10 |
+
avg=3
|
11 |
+
jit=1
|
12 |
+
|
13 |
+
dir=./conv_emformer_transducer_stateless2
|
14 |
+
|
15 |
+
./conv_emformer_transducer_stateless2/export.py \
|
16 |
+
--exp-dir $dir/exp-L-small \
|
17 |
+
--lang-dir ./data/lang_char \
|
18 |
+
--epoch $epoch \
|
19 |
+
--avg $avg \
|
20 |
+
--use-averaged-model 1 \
|
21 |
+
\
|
22 |
+
--encoder-dim 256 \
|
23 |
+
--nhead 4 \
|
24 |
+
--dim-feedforward 1024 \
|
25 |
+
--num-encoder-layers 5 \
|
26 |
+
--chunk-length 32 \
|
27 |
+
--cnn-module-kernel 31 \
|
28 |
+
--left-context-length 32 \
|
29 |
+
--right-context-length 8 \
|
30 |
+
--memory-size 32 \
|
31 |
+
--jit $jit
|
32 |
+
|
33 |
+
if [ $jit -eq 0 ]; then
|
34 |
+
pushd $dir/exp-L-small
|
35 |
+
mv pretrained.pt pretrained-epoch-$epoch-avg-$avg.pt
|
36 |
+
popd
|
37 |
+
fi
|
38 |
+
|
39 |
+
if [ $jit -eq 1 ]; then
|
40 |
+
pushd $dir/exp-L-small
|
41 |
+
mv cpu_jit.pt cpu_jit-epoch-$epoch-avg-$avg.pt
|
42 |
+
popd
|
43 |
+
fi
|