PyTorch深度学习实践代码 第四讲

#引入torch
import torch
#构建数据集
x_data = [1.0,2.0,3.0]
y_data = [2.0,4.0,6.0]
#用torch.Tensor创建一个权重变量，后面的1.0代表只有一个权重
w = torch.Tensor([1.0])
#权重w是需要计算梯度的，设它为真。
w.requires_grad = True
#构建模型
def forward(x):
    return x * w
#计算损失函数
def loss(x,y):
    y_pred = forward(x)
    return (y_pred - y) ** 2


#输出训练前的预测x=4,y=,后面加item()是要输出一个标量。
print("predict (before training)",4,forward(4).item())
#循环100次
for epoch in range(100):
    #用zip拼装送给x,y
    for x,y in zip(x_data,y_data):
        #计算损失，属于前馈过程
        l = loss(x,y)
        #损失函数的反向传播，计算当前梯度。释放计算图。
        l.backward()
        #item直接把梯度里面的数据拿出来，变成一个标量
        print('\tgrad:',x,y,w.grad.item())
        #权重的数据=初始权重的数据-学习率*权重梯度的数据
        w.data = w.data - 0.01 * w.grad.data
        #将权重的梯度的数据清零
        w.grad.data.zero_()
    #输出循环次数和损失,item变标量
    print("progress:",epoch,l.item())
#输出训练后预测，x=4,y=.item标量
print("predict (after training)",4,forward(4).item())