1.刘二大人第4讲
参考图引用于:PyTorch学习(三)--反向传播_陈同学爱吃方便面的博客-CSDN博客_反向传播pytorch
课后答案参考:
代码实现:
import torch
def BP_demo2(): x_data = [1.0, 2.0, 3.0] y_data = [2.0, 4.0, 6.0] epoch_list = [] l_list = [] w1 = torch.Tensor([1.0]) w1.requires_grad = True w2 = torch.Tensor([1.0]) w2.requires_grad = True b = torch.Tensor([1.0]) b.requires_grad = True def forward(x): return w1 * x ** 2 + w2 * x + b def loss(x, y): y_pred = forward(x) return (y_pred - y) ** 2 print("Predict (beforetraining):", 4, forward(4)) for epoch in range(100): for x, y in zip(x_data, y_data): l = loss(x, y) l.backward() print("\tgrad:", x, y, w1.grad.item(), w2.grad.item(), b.grad.item()) w1.data = w1.data - 0.01 * w1.grad.data w2.data = w2.data - 0.01 * w2.grad.data b.data = b.data - 0.01 * b.grad.data w1.grad.data.zero_() w2.grad.data.zero_() b.grad.data.zero_() print('epoch:', epoch, l.item()) print("predict(after training)", 4, forward(4).item())
if __name__=="__main__": # BP_demo1() BP_demo2()