Fix issues where -inf's are not properly dealt with.

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

Fix issues where -inf's are not properly dealt with. #3

Closed danpovey closed 1 year ago

danpovey commented 2 years ago

May not be fully tested. Simplifies the forward code by removing the "panic" mode and doing everything in log space; this is slower in the normal case, but since we will probably be dealing with -inf elements a lot, due to how the pruning code works, it will probably end up being faster after all.

pkufool commented 1 year ago

Can we close this PR? I think the fixes are already in master branch.

danpovey commented 1 year ago