yfyeung commited on
Commit
0c8bbdf
1 Parent(s): 03e3e37
Files changed (5) hide show
  1. exp/cpu_jit.pt +3 -0
  2. exp/export.sh +10 -0
  3. exp/pretrained.pt +3 -0
  4. exp/pretrained.sh +8 -0
  5. exp/run.sh +11 -0
exp/cpu_jit.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed1f79e45a86d360e7a140667e8d3f90ee8a5ea87a88d9d9618cdcce17a11e98
3
+ size 358526334
exp/export.sh ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ export CUDA_VISIBLE_DEVICES=
4
+
5
+ ./pruned_transducer_stateless7/export.py \
6
+ --exp-dir ./pruned_transducer_stateless7/exp \
7
+ --bpe-model data/en/lang_bpe_500/bpe.model \
8
+ --epoch 30 \
9
+ --avg 5 \
10
+ --jit 0
exp/pretrained.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c8ff8a26e7a53a4ca7aa63ea474c3b75a5b26209443dd6aa0897c591d7c7596
3
+ size 281768753
exp/pretrained.sh ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ ./pruned_transducer_stateless7/pretrained.py \
4
+ --checkpoint ./pruned_transducer_stateless7/exp/pretrained.pt \
5
+ --bpe-model ./data/en/lang_bpe_500/bpe.model \
6
+ --method greedy_search \
7
+ ./test_wavs/1089-134686-0001.wav \
8
+ ./test_wavs/1221-135766-0001.wav
exp/run.sh ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+
3
+ export CUDA_VISIBLE_DEVICES="0,1,2,3"
4
+
5
+ ./pruned_transducer_stateless7/train.py \
6
+ --world-size 4 \
7
+ --num-epochs 30 \
8
+ --start-epoch 1 \
9
+ --use-fp16 1 \
10
+ --exp-dir pruned_transducer_stateless7/exp \
11
+ --max-duration 550