search

HawkAaron / RNN-Transducer

MXNet implementation of RNN Transducer (Graves 2012): Sequence Transduction with Recurrent Neural Networks
136 stars 31 forks source link

Where does this come from? self.loss = gluon.loss.RNNTLoss(blank_label=blank) #2

Closed rzcwade closed 6 years ago

rzcwade commented 6 years ago

Hello,

I am trying to run your code and I discovered that you used gluon.loss.RNNTLoss() which does not exist. Could you clarify on this function? Did you mean from warprnnt_pytorch import RNNTLoss?

Thanks.

HawkAaron commented 6 years ago

This is a mxnet version, so you need to use mxnet rnnt loss, which is in https://github.com/HawkAaron/mxnet-transducer/tree/add_network

rzcwade commented 6 years ago

Thanks, @HawkAaron !