What is the difference between CuDNNLSTM and LSTM in Keras?

Why don’t you try it out for yourself and see?
In my case, training a model with LSTM took 10mins 30seconds.
Simply switching the call from LSTM() to CuDNNLSTM() took less than a minute.

I also noticed that switching to CuDNNLSTM() speeds up model.evaluate() and model.predict() substantially as well.

Leave a Comment