导入RMSprop、Adam出错

问题描述:通过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)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。