【问题标题】:AttributeError: module 'torch.optim' has no attribute 'RMSProp'AttributeError:模块“torch.optim”没有属性“RMSProp”
【发布时间】:2020-06-20 02:55:24
【问题描述】:

尝试将 RMSProp 优化器与 PyTorch 一起使用时出现以下错误:

AttributeError: module 'torch.optim' has no attribute 'RMSProp'

代码:

import torch as T
import torch.nn as nn
import torch.optim as optim

class DeepQNetwork(nn.Module):

  def __init__(self, alpha, ...):
    super(DeepQNetwork, self).__init__()
    ...
    self.optimizer = optim.RMSProp(self.parameters(), lr=alpha)
    ...

PyTorch 版本为1.5.1,Python 版本为3.6torch.optim 及其优化器包括 RMSProp 有一个文档,但 PyCharm 只建议 Adam 和 SGD,而且似乎所有其他优化器都不见了。

有人有想法吗?我在互联网上没有找到任何东西,它开始让我发疯。

Suggesstion from PyCharm

【问题讨论】:

  • 运行代码时python会抛出错误吗?
  • Python 抛出上述的 AttributeError。

标签: python optimization pytorch


【解决方案1】:

RMSprop(如文档中所示)而不是 RMSProp。所以,这只是一个错字。

【讨论】:

  • 谢谢,我很困惑 PyCharm 只建议了 Adam 和 SGD,我没有注意到错字。
猜你喜欢
  • 2021-06-21
  • 1970-01-01
  • 1970-01-01
  • 2018-04-14
  • 2019-02-18
  • 1970-01-01
  • 2020-01-01
  • 2019-07-20
相关资源
最近更新 更多