Downloads · 30 days
0
icefall-asr-librispeech-pruned-stateless-streaming-conformer-rnnt4-2022-06-10
pkufool/icefall-asr-librispeech-pruned-stateless-streaming-conformer-rnnt4-2022-06-10
icefall-asr-librispeech-pruned-stateless-streaming-conformer-rnnt4-2022-06-10 is a machine learning model from pkufool. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
The pretrained model (prunedtransducerstateless4) in https://github.com/k2-fsa/icefall/pull/380
Access
Public
Updated Jul 28, 2022
Repo size
1.3 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt1.6 GB · 93%
At a glance
- License
- apache-2.0
- Access
- Public
- Created
- Jun 9, 2022
- Updated
- Jul 28, 2022
- SHA
- 6a86136f
Model card
From the Hugging Face model README
The pretrained model (pruned_transducer_stateless4) in https://github.com/k2-fsa/icefall/pull/380
training
#!/usr/bin/env bash
set -x
K2_ROOT=/path/to/k2
ICEFALL=/path/to/icefall
export PYTHONPATH=$K2_ROOT/k2/python:$PYTHONPATH
export PYTHONPATH=$K2_ROOT/build/lib:$PYTHONPATH
export PYTHONPATH=$ICEFALL:$PYTHONPATH
export CUDA_VISIBLE_DEVICES="0,1,2,3"
./pruned_transducer_stateless4/train.py \
--exp-dir pruned_transducer_stateless4/exp \
--full-libri 1 \
--dynamic-chunk-training 1 \
--short-chunk-size 32 \
--num-left-chunks 4 \
--causal-convolution 1 \
--max-duration 300 \
--world-size 4 \
--start-epoch 1 \
--num-epochs 30
decoding
simulate streaming
#!/usr/bin/env bash
set -x
K2_ROOT=/path/to/k2
ICEFALL=/path/to/icefall
export PYTHONPATH=$K2_ROOT/k2/python:$PYTHONPATH
export PYTHONPATH=$K2_ROOT/build/lib:$PYTHONPATH
export PYTHONPATH=$ICEFALL:$PYTHONPATH
export CUDA_VISIBLE_DEVICES="0"
for size in 1 2 4 8 16 32; do
for left in 32 64 -1; do
./pruned_transducer_stateless4/decode.py \
--simulate-streaming 1 \
--decode-chunk-size ${size} \
--left-context ${left} \
--causal-convolution 1 \
--use-averaged-model 1 \
--epoch 29 \
--avg 6 \
--exp-dir ./pruned_transducer_stateless4/exp \
--max-sym-per-frame 1 \
--max-duration 1000 \
--decoding-method greedy_search
done
done
streaming
#!/usr/bin/env bash
set -x
K2_ROOT=/path/to/k2
ICEFALL=/path/to/icefall
export PYTHONPATH=$K2_ROOT/k2/python:$PYTHONPATH
export PYTHONPATH=$K2_ROOT/build/lib:$PYTHONPATH
export PYTHONPATH=$ICEFALL:$PYTHONPATH
export CUDA_VISIBLE_DEVICES="0"
#left_context=32
#chunk_size=8
left_context=64
chunk_size=16
for right in 0 2 4 8; do
./pruned_transducer_stateless4/streaming_decode.py \
--left-context ${left_context} \
--decode-chunk-size ${chunk_size} \
--right-context ${right} \
--exp-dir ./pruned_transducer_stateless4/exp \
--use-averaged-model 1 \
--epoch 29 \
--avg 6 \
--num-decode-streams 1000
done
export
for pretrained.pt
python pruned_transducer_stateless4/export.py \
--exp-dir ./pruned_transducer_stateless4/exp \
--epoch 29 \
--avg 6 \
--streaming-model 1 \
--causal-convolution 1
for cpu_jit.pt
python pruned_transducer_stateless4/export.py \
--exp-dir ./pruned_transducer_stateless4/exp \
--epoch 29 \
--avg 6 \
--streaming-model 1 \
--causal-convolution 1 \
--jit 1
- License
- apache-2.0
- Created
- Jun 9, 2022
- Updated
- Jul 28, 2022