mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-10 10:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
icefall/egs/librispeech/ASR/transducer
History
Fangjun Kuang 5b6699a835
Minor fixes to the RNN-T Conformer model (#152) 
* Disable weight decay.

* Remove input feature batchnorm..

* Replace BatchNorm in the Conformer model with LayerNorm.

* Use tanh in the joint network.

* Remove sos ID.

* Reduce the number of decoder layers from 4 to 2.

* Minor fixes.

* Fix typos.
2021-12-23 13:54:25 +08:00
..
__init__.py
Apply layer normalization to the output of each gate in LSTM/GRU. (#139)
2021-12-07 18:38:03 +08:00
asr_datamodule.py
RNN-T Conformer training for LibriSpeech (#143)
2021-12-18 07:42:51 +08:00
beam_search.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
conformer.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
decode.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
decoder.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
encoder_interface.py
RNN-T Conformer training for LibriSpeech (#143)
2021-12-18 07:42:51 +08:00
export.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
joiner.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
model.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
pretrained.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
README.md
RNN-T Conformer training for LibriSpeech (#143)
2021-12-18 07:42:51 +08:00
rnn.py
RNN-T Conformer training for LibriSpeech (#143)
2021-12-18 07:42:51 +08:00
subsampling.py
RNN-T Conformer training for LibriSpeech (#143)
2021-12-18 07:42:51 +08:00
test_conformer.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
test_decoder.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
test_joiner.py
RNN-T Conformer training for LibriSpeech (#143)
2021-12-18 07:42:51 +08:00
test_rnn.py
RNN-T Conformer training for LibriSpeech (#143)
2021-12-18 07:42:51 +08:00
test_transducer.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
test_transformer.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
train.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
transformer.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00

README.md

Introduction

The encoder consists of Conformer layers in this folder. You can use the following command to start the training:

cd egs/librispeech/ASR

export CUDA_VISIBLE_DEVICES="0,1,2,3"

./transducer/train.py \
  --world-size 4 \
  --num-epochs 30 \
  --start-epoch 0 \
  --exp-dir transducer/exp \
  --full-libri 1 \
  --max-duration 250 \
  --lr-factor 2.5