【Pytorch深度学习开发实践学习】B站刘二大人课程笔记整理lecture04反向传播-Toy模板网

这篇具有很好参考价值的文章主要介绍了【Pytorch深度学习开发实践学习】B站刘二大人课程笔记整理lecture04反向传播。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

lecture04反向传播
课程网址

Pytorch深度学习实践

部分课件内容：
【Pytorch深度学习开发实践学习】B站刘二大人课程笔记整理lecture04反向传播,Pytorch深度学习开发实践学习,深度学习,pytorch,学习

【Pytorch深度学习开发实践学习】B站刘二大人课程笔记整理lecture04反向传播,Pytorch深度学习开发实践学习,深度学习,pytorch,学习

import torch

x_data =[1.0,2.0,3.0]
y_data =[2.0,4.0,6.0]
w = torch.tensor([1.0])
w.requires_grad = True

def forward(x):
    return x*w

def loss(x,y):
    y_pred = forward(x)
    return (y_pred-y)**2

print('Predicted(before training)',4,forward(4).item())

for epoch in range(100):
    for x,y in zip(x_data,y_data):
        l = loss(x,y)
        l.backward()
        print("\tgrad:",x,y,w.grad.item())
        w.data = w.data - 0.01 * w.grad.item()
        w.grad.zero_()
    print('Epoch:', epoch,'w:',w.item(), 'Loss:', l.item())

print('Predicted(after training)',4,forward(4).item())