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 20 \
+    --avg 5 \
+    --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 20 \
+  --avg 5 \
+  --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 20 \
+  --avg 5 \
+  --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,22 @@
+#!/usr/bin/env bash
+
+export CUDA_VISIBLE_DEVICES="0,1"
+
+git lfs install
+git clone https://huggingface.co/csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11
+
+./pruned_transducer_stateless7/finetune.py \
+  --world-size 2 \
+  --num-epochs 20 \
+  --start-epoch 1 \
+  --exp-dir pruned_transducer_stateless7/exp \
+  --subset S \
+  --use-fp16 1 \
+  --base-lr 0.005 \
+  --lr-epochs 100 \
+  --lr-batches 100000 \
+  --bpe-model icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11/data/lang_bpe_500/bpe.model \
+  --do-finetune 1 \
+  --use-mux 1 \
+  --finetune-ckpt icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11/exp/pretrained.pt \
+  --max-duration 500