6 Commits

Author SHA1 Message Date
Fangjun Kuang
3e2dbc9ab5 Minor fixes. 2022-05-23 18:18:04 +08:00
Fangjun Kuang
bf3df442c6 Merge remote-tracking branch 'dan/master' into rnnt-lstm-2022-04-21 2022-05-23 17:33:26 +08:00
Fangjun Kuang
f6ce135608
Various fixes to support torch script. (#371)
* Various fixes to support torch script.

* Add tests to ensure that the model is torch scriptable.

* Update tests.
2022-05-16 21:46:59 +08:00
Fangjun Kuang
d11e01e190 Minor fixes. 2022-05-10 22:06:34 +08:00
Fangjun Kuang
52b3ed2920 Use a stateless decoder for transducer_lstm. 2022-04-21 14:05:38 +08:00
Fangjun Kuang
1d44da845b
RNN-T Conformer training for LibriSpeech (#143)
* Begin to add RNN-T training for librispeech.

* Copy files from conformer_ctc.

Will edit it.

* Use conformer/transformer model as encoder.

* Begin to add training script.

* Add training code.

* Remove long utterances to avoid OOM when a large max_duraiton is used.

* Begin to add decoding script.

* Add decoding script.

* Minor fixes.

* Add beam search.

* Use LSTM layers for the encoder.

Need more tunings.

* Use stateless decoder.

* Minor fixes to make it ready for merge.

* Fix README.

* Update RESULT.md to include RNN-T Conformer.

* Minor fixes.

* Fix tests.

* Minor fixes.

* Minor fixes.

* Fix tests.
2021-12-18 07:42:51 +08:00