Commit
•
57bdcfe
1
Parent(s):
60c1fc9
Add export script.
Browse files- exp/export.sh +18 -0
exp/export.sh
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env bash
|
| 2 |
+
|
| 3 |
+
set -e
|
| 4 |
+
|
| 5 |
+
export CUDA_VISIBLE_DEVICES=0
|
| 6 |
+
|
| 7 |
+
. path.sh
|
| 8 |
+
epoch=38
|
| 9 |
+
avg=10
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
./pruned_transducer_stateless2/export.py \
|
| 13 |
+
--exp-dir ./pruned_transducer_stateless2/exp \
|
| 14 |
+
--bpe-model data/lang_bpe_500/bpe.model \
|
| 15 |
+
--epoch $epoch \
|
| 16 |
+
--avg $avg
|
| 17 |
+
|
| 18 |
+
mv -v ./pruned_transducer_stateless2/exp/pretrained.pt ./pruned_transducer_stateless2/exp/pretrained-epoch-${epoch}-avg-${avg}.pt
|