Skip to content

Conversation

GameLogist
Copy link

I believe CuDNNLSTM will be beneficial for the model.

@TheErk
Copy link

TheErk commented Oct 3, 2018

I tested this on my side and CuDNNLSTM seems to work as a drop-in replacement for LSTM.
This only works for keras tensorflow backend though.

That said in my current test case using CuDNNLSTM instead of LSTM make learning phase go 3 times faster. I don't took time to thoroughly cross check results between the 2 variants after learning.

Perhaps one could programmatically check backend and use CuDNNLSTM whenever

keras.backend.backend()=='tensorflow'

@GameLogist thank you for your suggestion I'll use it when possible.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants