Модуль tenensflow.python.keras.api._v2.keras.layers не имеет атрибута CuDNNLSTM.

Question

Когда я пишу tf.keras.layers.LSTM, я получаю предупреждение

Note that this layer is not optimized for performance. Please use tf.keras.layers.CuDNNLSTM for better performance on GPU.

Но когда я меняю слой на tf.keras.layers.CuDNNLSTM, я получаю ошибку

AttributeError: module 'tensorflow.python.keras.api._v2.keras.layers' has no attribute 'CuDNNLSTM'

Версия Tensorflow 2.0.0-alpha0, версия Keras 2.2.4-tf.

Как я могу исправить эту проблему?

Answer 1

В общем, в TensorFlow 2.0 мы просто должны использовать:

tf.keras.layers.LSTM

, который, несмотря на предупреждение, будет использовать графический процессор.

Предупреждающее сообщение неправильно существовало в версии 2.0.0-alpha0, но с тех пор было удалено в 2.0.0-beta1

Если по какой-то причине вам нужна оригинальная реализация tf.keras.layers.CuDNNLSTM, тогда вы можете использовать tf.compat.v1.keras.layers.CuDNNLSTM, но это будет крайний случай.