Upload folder using huggingface_hub

exp/decode.sh

@@ -0,0 +1,28 @@
+# simulated streaming decoding
+export CUDA_VISIBLE_DEVICES="0"
+for m in greedy_search modified_beam_search fast_beam_search; do
+  ./zipformer/decode.py \
+      --epoch 30 \
+      --avg 8 \
+      --exp-dir zipformer/exp-causal/ \
+      --max-duration 600 \
+      --causal 1 \
+      --chunk-size 16 \
+      --left-context-frames 128 \
+      --decoding-method $m
+done
+
+# chunk-wise streaming decoding
+export CUDA_VISIBLE_DEVICES="0"
+for m in greedy_search modified_beam_search fast_beam_search; do
+  ./zipformer/streaming_decode.py \
+      --epoch 30 \
+      --avg 8 \
+      --exp-dir zipformer/exp-causal/ \
+      --causal 1 \
+      --chunk-size 16 \
+      --left-context-frames 128 \
+      --num-decode-streams 1000 \
+      --decoding-method $m
+done
+

exp/epoch-30.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:49c56b6984d25caedd1b017e2ead5c61f65c143d7353b0f62b0ba7949484a3ad
+size 135

exp/export.sh

@@ -0,0 +1,11 @@
+./zipformer/export.py \
+  --exp-dir ./zipformer/exp-causal \
+  --bpe-model data/lang_bpe_500/bpe.model \
+  --causal 1 \
+  --chunk-size 16 \
+  --left-context-frames 128 \
+  --epoch 30 \
+  --avg 8 \
+  --jit 1
+
+

exp/jit_pretrained_streaming.sh

@@ -0,0 +1,10 @@
+export CUDA_VISIBLE_DEVICES="0"
+
+./zipformer/jit_pretrained_streaming.py \
+  --nn-model-filename ./zipformer/exp-causal/jit_script_chunk_16_left_128.pt \
+  --bpe-model ./data/lang_bpe_500/bpe.model \
+  test_wavs/1221-135766-0001.wav
+  # test_wavs/1089-134686-0001.wav
+  # test_wavs/1221-135766-0002.wav
+
+

exp/jit_script_chunk_16_left_128.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f90279ab5eb351f67f2a1ba61990b78706752bdf3656fef24a4e405e1846efa
+size 267277530

exp/pretrained.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:28164e2ee15a6c04b5206422cbfb8b795fbc01c4d04f9d85a906b4eb5d953647
+size 134

exp/pretrained.sh

@@ -0,0 +1,13 @@
+export CUDA_VISIBLE_DEVICES=0
+
+./zipformer/pretrained.py \
+  --checkpoint ./zipformer/exp-causal/pretrained.pt \
+  --bpe-model ./data/lang_bpe_500/bpe.model \
+  --method greedy_search \
+  --causal 1 \
+  --chunk-size 16 \
+  --left-context-frames 128 \
+  test_wavs/1089-134686-0001.wav \
+  test_wavs/1221-135766-0001.wav \
+  test_wavs/1221-135766-0002.wav
+

exp/train.sh

@@ -0,0 +1,13 @@
+export CUDA_VISIBLE_DEVICES="0,1,2,3"
+./zipformer/train.py \
+  --world-size 4 \
+  --num-epochs 30 \
+  --start-epoch 1 \
+  --use-fp16 1 \
+  --exp-dir zipformer/exp-causal \
+  --causal 1 \
+  --full-libri 1 \
+  --max-duration 1000 \
+  --master-port 12345
+
+