icefall/egs/tedlium3/ASR/transducer_stateless
Yifan Yang 5dfc3ed7f9
Fix buffer size of DynamicBucketingSampler (#1468)
* Fix buffer size

* Fix for flake8

---------

Co-authored-by: yifanyeung <yifanyeung@yifanyeung.local>
2024-01-21 02:10:42 +08:00
..
2022-11-17 09:42:17 -05:00
2023-03-08 22:56:04 +08:00
2022-11-17 09:42:17 -05:00
2022-03-18 11:39:06 +08:00
2023-09-26 15:41:39 +08:00
2022-03-18 11:39:06 +08:00
2022-11-17 14:18:05 -05:00

Introduction

The decoder, i.e., the prediction network, is from https://ieeexplore.ieee.org/stamp/stamp.jsp?arnumber=9054419 (Rnn-Transducer with Stateless Prediction Network)

You can use the following command to start the training:

cd egs/tedlium3/ASR

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