TensorFlowASR icon indicating copy to clipboard operation
TensorFlowASR copied to clipboard

Published 20 hours ago • verified publisher icon TensorSpeech

RNNT loss

Open liuyibox opened this issue 3 years ago • 0 comments

I have tried to use warprnnt-tensorflow to compute the loss, however, it turns out warprnnt-tensorflow binding does not support our GPU (A30). So I think I need to focus on the computing the loss using tensorflow rnnt loss.

The code to compute the warprnnt is quite complicated here: https://github.com/TensorSpeech/TensorFlowASR/blob/main/tensorflow_asr/losses/rnnt_loss.py#L236. Can we have any resources to better understand this loss and modify the code? Thanks.

liuyibox avatar Sep 24 '22 17:09 liuyibox