【发布时间】: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.6。 torch.optim 及其优化器包括 RMSProp 有一个文档,但 PyCharm 只建议 Adam 和 SGD,而且似乎所有其他优化器都不见了。
有人有想法吗?我在互联网上没有找到任何东西,它开始让我发疯。
【问题讨论】:
-
运行代码时python会抛出错误吗?
-
Python 抛出上述的 AttributeError。
标签: python optimization pytorch