What is the difference between CuDNNLSTM and LSTM in Keras?

前端 未结 3 1330
失恋的感觉
失恋的感觉 2021-01-30 12:57

In Keras, the high-level deep learning library, there are multiple types of recurrent layers; these include LSTM (Long short term memory) and CuD

3条回答
  •  野性不改
    2021-01-30 13:47

    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.

提交回复
热议问题