当我写作时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。
我该如何解决这个问题?
一般来说,在 TensorFlow 2.0 中我们应该只使用:
tf.keras.layers.LSTM
尽管有警告,will使用 GPU。
警告消息不正确地存在于2.0.0-alpha0
版本但已被删除2.0.0-beta1
如果由于某种原因您特别需要原始实现tf.keras.layers.CuDNNLSTM
那么你可以使用tf.compat.v1.keras.layers.CuDNNLSTM
但这将是一种边缘情况。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)