Element-Research / rnn

Recurrent Neural Network library for Torch7's nn
BSD 3-Clause "New" or "Revised" License
939 stars 313 forks source link

Using rnn from torch-rnn for cuDNN #365

Open suryabhupa opened 7 years ago

suryabhupa commented 7 years ago

The interface for rnn's LSTM updateOutput() seems slightly different from torch-rnn's updateOutput(). Is there a nice analog for changing code from using one to the other?

More directly, my code currently uses torch-rnn's LSTM implementation but I would like to switch to rnn's LSTM implementation because it seems to be able to converted to cudnn.LSTM more easily. I've seen this fork: https://github.com/ngimel/torch-rnn, but converting to cudnn doesn't seem to work.

Does anyone know any work arounds?