问题描述:通过pip install tensorflow后。执行下面代码出现ImportError: cannot import name RMSprop错误
from tensorflow.python.keras.optimizers import RMSprop
from tensorflow.python.keras.optimizers import RMSprop
原因:安装的是tensorflow版本较新,同样tensorflow内的keras版本或许也变了?
解决:
- 查看optimizers.py文件。发现:
from tensorflow.python.keras.optimizer_v2 import adam as adam_v2
from tensorflow.python.keras.optimizer_v2 import adamax as adamax_v2
.....
all_classes = {
'adadelta': adadelta_v2.Adadelta,
'adagrad': adagrad_v2.Adagrad,
'adam': adam_v2.Adam,
'adamax': adamax_v2.Adamax,
'nadam': nadam_v2.Nadam,
-
解决方案:
根据下面的代码引用from tensorflow.python.keras.optimizers import adam_v2 from tensorflow.python.keras.optimizers import rmsprop_v2
调用adam优化器
使用
optimizer =adam_v2.Adam(lr=1e-3)
而不是
optimizer = Adam(lr=1e-3)