In Keras
, the high-level deep learning library, there are multiple types of recurrent layers; these include LSTM
(Long short term memory) and CuD
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.