Upload 6 files

exp/decode.sh

@@ -0,0 +1,11 @@
+#!/usr/bin/env bash
+
+export CUDA_VISIBLE_DEVICES=
+
+# greedy search
+./pruned_transducer_stateless7/decode.py \
+    --epoch 30 \
+    --avg 4 \
+    --exp-dir ./pruned_transducer_stateless7/exp \
+    --max-duration 600 \
+    --decoding-method greedy_search

exp/export-onnx.sh

@@ -0,0 +1,10 @@
+#!/usr/bin/env bash
+
+export CUDA_VISIBLE_DEVICES=
+
+./pruned_transducer_stateless7/export-onnx.py \
+  --bpe-model ./data/lang_bpe_500/bpe.model \
+  --use-averaged-model 1 \
+  --epoch 30 \
+  --avg 4 \
+  --exp-dir ./pruned_transducer_stateless7/exp

exp/export.sh

@@ -0,0 +1,10 @@
+#!/usr/bin/env bash
+
+export CUDA_VISIBLE_DEVICES=
+
+./pruned_transducer_stateless7/export.py \
+  --exp-dir ./pruned_transducer_stateless7/exp \
+  --bpe-model data/lang_bpe_500/bpe.model \
+  --epoch 30 \
+  --avg 4 \
+  --jit 1

exp/onnx_pretrained.sh

@@ -0,0 +1,11 @@
+#!/usr/bin/env bash
+
+
+./pruned_transducer_stateless7/onnx_pretrained.py \
+  --encoder-model-filename ./pruned_transducer_stateless7/exp/encoder-epoch-9999-avg-1.onnx \
+  --decoder-model-filename ./pruned_transducer_stateless7/exp/decoder-epoch-9999-avg-1.onnx \
+  --joiner-model-filename ./pruned_transducer_stateless7/exp/joiner-epoch-9999-avg-1.onnx \
+  --tokens ./data/lang_bpe_500/tokens.txt \
+  ./test_wavs/1089-134686-0001.wav \
+  ./test_wavs/1221-135766-0001.wav \
+  ./test_wavs/1221-135766-0002.wav

exp/pretrained.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+
+./pruned_transducer_stateless7/pretrained.py \
+  --checkpoint ./pruned_transducer_stateless7/exp/pretrained.pt \
+  --bpe-model ./data/lang_bpe_500/bpe.model \
+  --method greedy_search \
+  ./test_wavs/1089-134686-0001.wav \
+  ./test_wavs/1221-135766-0001.wav

exp/run.sh

@@ -0,0 +1,12 @@
+#!/usr/bin/env bash
+
+export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
+
+./pruned_transducer_stateless7/train.py \
+  --world-size 8 \
+  --num-epochs 30 \
+  --start-epoch 1 \
+  --exp-dir pruned_transducer_stateless7/exp \
+  --use-multidataset 1 \
+  --use-fp16 1 \
+  --max-duration 750