头歌Pytorch之优化第1关：如何使用optimize

这题很傻逼，输出就得用他这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 *********/

后面慢慢更