这题很傻逼,输出就得用他这optimizer.param_groups[0]['lr'],测试代码有检测,自己写的时候是用循环遍历字典,再输出lr键值,一直报错,后面看测试代码才发现检测机制是param_groups[0]['lr'],真捞。

import torch
from torch.autograd import Variable

# Linear Regression Model
class LinearRegression(nn.Module):
    def __init__(self):
        super(LinearRegression, self).__init__()
        self.linear = nn.Linear(2, 2)  # input and output is 2 dimension
    
    def forward(self, x):
        out = self.linear(x)
        return out
model = LinearRegression()


#/********** Begin *********/

#声明一个 SGD优化器 optimizer,传入参数
optimizer= torch.optim.SGD(model.parameters(), lr = 0.1)
#利用optimizer.param_groups查看优化器的各项参数并输出lr的值。

print('lr:',optimizer.param_groups[0]['lr'])

#/********** End *********/

后面慢慢更

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐