原代码:
from keras.layers import CuDNNGRU, CuDNNLSTM, SpatialDropout1D
报错:
ImportError:cannot import name ‘CuDNNGRU‘
分开引入:
from keras.layers import CuDNNGRU
from keras.layers import CuDNNLSTM
from keras.layers import SpatialDropout1D
报错:
ImportError:cannot import name ‘CuDNNGRU‘
ImportError:cannot import name ‘CuDNNLSTM‘
SpatialDropout1D正常引入
最后改为:
from tensorflow.compat.v1.keras.layers import CuDNNGRU as GRU
from tensorflow.compat.v1.keras.layers import CuDNNLSTM as LSTM
from keras.layers import SpatialDropout1D
运行正常。