search

k2-fsa / fast_rnnt

A torch implementation of a recursion which turns out to be useful for RNN-T.
Other
136 stars 22 forks source link

Add aux version of rnnt loss, allows to have "lm-only" and "am-only" … #1

Closed danpovey closed 2 years ago

danpovey commented 2 years ago

…versions of the loss

which may add regularization, and also ensure that the ILM probs make sense independently.