Downloads · 30 days
0
luomingshuang/icefall_asr_tedlium3_transducer_stateless
icefall_asr_tedlium3_transducer_stateless is a machine learning model from luomingshuang. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Note: This recipe is trained with the codes from this PR https://github.com/k2-fsa/icefall/pull/233 And the SpecAugment codes from this PR https://github.com/lhotse-speech/lhotse/pull/604.
Downloads · 30 days
0
Access
Public
Updated Jul 20, 2022
Repo size
2.7 GB
Likes
0
Public
Click a slice to open those files.
.pt2.4 GB · 100%
From the Hugging Face model README
Note: This recipe is trained with the codes from this PR https://github.com/k2-fsa/icefall/pull/233 And the SpecAugment codes from this PR https://github.com/lhotse-speech/lhotse/pull/604.
The model was trained on full TEDLium3 with the scripts in icefall.
The main repositories are list below, we will update the training and decoding scripts with the update of version.
k2: https://github.com/k2-fsa/k2
icefall: https://github.com/k2-fsa/icefall
lhotse: https://github.com/lhotse-speech/lhotse
git clone https://github.com/k2-fsa/icefall
cd icefall
cd egs/tedlium3/ASR
bash ./prepare.sh
export CUDA_VISIBLE_DEVICES="0,1,2,3"
./transducer_stateless/train.py \
--world-size 4 \
--num-epochs 30 \
--start-epoch 0 \
--exp-dir transducer_stateless/exp \
--max-duration 300
The decoding results (WER%) on TEDLium3 (dev and test) are listed below, we got this result by averaging models from epoch 19 to 29. The WERs are
| dev | test | comment | |
|---|---|---|---|
| greedy search | 7.19 | 6.70 | --epoch 29, --avg 11, --max-duration 100 |
| beam search (beam size 4) | 7.02 | 6.36 | --epoch 29, --avg 11, --max-duration 100 |
| modified beam search (beam size 4) | 6.91 | 6.33 | --epoch 29, --avg 11, --max-duration 100 |