mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity
icefall/egs/librispeech/ASR/transducer
History
Fangjun Kuang fba5e67d5e
Fix CI tests. (#1974) 
- Introduce unified AMP helpers (create_grad_scaler, torch_autocast) to handle 
  deprecations in PyTorch ≥2.3.0

- Replace direct uses of torch.cuda.amp.GradScaler and torch.cuda.amp.autocast 
  with the new utilities across all training and inference scripts

- Update all torch.load calls to include weights_only=False for compatibility with 
  newer PyTorch versions
2025-07-01 13:47:55 +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
apply black on all files
2022-11-17 09:42:17 -05:00
conformer.py
Fix joiner (#234)
2022-03-02 16:41:14 +08:00
decode.py
Fix style issues. (#937)
2023-03-08 22:56:04 +08:00
decoder.py
Fix some typos. (#329)
2022-04-22 15:54:59 +08:00
encoder_interface.py
Fix joiner (#234)
2022-03-02 16:41:14 +08:00
export.py
Use tokens.txt to replace bpe.model (#1162)
2023-08-12 16:53:59 +08:00
joiner.py
Minor fixes to the RNN-T Conformer model (#152)
2021-12-23 13:54:25 +08:00
model.py
Fix torch.nn.Embedding error for torch below 1.8.0 (#198)
2022-02-06 21:59:54 +08:00
pretrained.py
Fix CI tests. (#1974)
2025-07-01 13:47:55 +08:00
README.md
RNN-T Conformer training for LibriSpeech (#143)
2021-12-18 07:42:51 +08:00
rnn.py
apply black on all files
2022-11-17 09:42:17 -05: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
Ban the test_rnn.py in ci-test (#949)
2023-03-15 22:02:20 +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
formatted the entire LibriSpeech recipe (#1270)
2023-09-24 17:31:01 +08:00
transformer.py
Fix joiner (#234)
2022-03-02 16:41:14 +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