Pytorch 之优化

第1关：如何使用optimizer

本关卡要求同学们声明一个 SGD 优化器 optimizer，按照要求传入优化器的参数。同时，为了便于观察，利用optimizer.param_groups方法查看优化器的各项参数并输出。

import torch.nn as nn
import torch.optim
import torch
from torch.autograd import Variable
# Linear Regression Model
class LinearRegression(nn.Module):
    d
版权声明：本文为Junds0原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。