ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

【PyTorch基础教程4】反向传播与计算图(学不会来打我啊)

2021-10-17 21:03:00  阅读:205  来源: 互联网

标签:loss 基础教程 grad 0.01 item PyTorch 反向 计算 data


学习总结

(1)loss实际在构建计算图,每次运行完后计算图就释放了。
(2)Tensor的Grad也是一个Tensor。更新权重w.data = w.data - 0.01 * w.grad.data的0.01乘那坨其实是在建立计算图,而我们这里要乘0.01 * grad.data,这样是不会建立计算图的(并不希望修改权重w,后面还有求梯度)。
(3)下面的w.grad.item()是直接把w.grad的数值取出,变成一个标量(也是为了防止产生计算图)。总之,牢记权重更新过程中要使用data

文章目录

一、基础回顾

1.1 正向传递

在这里插入图片描述

1.2 反向传播

在这里插入图片描述

1.3 举栗子

现在以 f = x ⋅ ω f=x \cdot \omega f=x⋅ω 为例:

(1)正向传递

在这里插入图片描述

(2)反向传播

在这里插入图片描述
注意虽然这里的 ∂ L ∂ x = ∂ L ∂ z ⋅ ∂ z ∂ x \frac{\partial L}{\partial x}=\frac{\partial L}{\partial z} \cdot \frac{\partial z}{\partial x} ∂x∂L​=∂z∂L​⋅∂x∂z​不求也可以,但是在pytorch是会求出来的(因为如果是多层,则需要用到该中间层求得的的梯度)。

二、计算图

2.1 线性模型的计算图

在这里插入图片描述
练习:
在这里插入图片描述

三、代码实战

# -*- coding: utf-8 -*-
"""
Created on Sun Oct 17 19:39:32 2021

@author: 86493
"""
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

# 这里使用SGD
def loss(x, y):
    y_pred = forward(x)
    return (y_pred - y) ** 2

print("predict (before training)", 4,
      forward(4).item())

# 训练过程,SGD
for epoch in range(100):
    for x, y in zip(x_data, y_data):
        # 向前传播,计算loss
        l = loss(x, y)
        # 计算requires_grad为true的tensor的梯度
        l.backward()
        print('\tgrad:', x, y, w.grad.item())
        w.data = w.data - 0.01 * w.grad.data
        # 反向传播后grad会被重复计算,所以记得清零梯度
        w.grad.data.zero_()
        
    print("progress:", epoch, l.item())
    
print("predict (after training)", 4, 
      forward(4).item())   

注意:
(1)loss实际在构建计算图,每次运行完后计算图就释放了。
(2)Tensor的Grad也是一个Tensor。更新权重w.data = w.data - 0.01 * w.grad.data的0.01乘那坨其实是在建立计算图,而我们这里要乘0.01 * grad.data,这样是不会建立计算图的(并不希望修改权重w,后面还有求梯度)。
(3)w.grad.item()是直接把w.grad的数值取出,变成一个标量(也是为了防止产生计算图)。总之,牢记权重更新过程中要使用data
在这里插入图片描述
(4)如果不像上面计算一个样本的loss,想算所有样本的loss(cost),然后就加上sum += l,注意此时sum是关于张量 l l l 的一个计算图,又未对sumbackward操作,随着l越加越多会导致内存爆炸。
正确做法:sum += l.item(),别把损失直接加到sum里面。
Tensor在做加法运算时会构建计算图
(5)backward后的梯度一定要记得清零w.grad.data.zero()
在这里插入图片描述
(6)训练过程:先计算loss损失值,然后backward反向传播,现在就有了梯度了。通过梯度下降更新参数:
在这里插入图片描述

四、作业

在这里插入图片描述


Reference

(1)PyTorch 深度学习实践 第10讲,刘二系列
(2)b站视频:https://www.bilibili.com/video/BV1Y7411d7Ys?p=10
(3)官方文档:https://pytorch.org/docs/stable/_modules/torch/nn/modules/conv.html#Conv2d
(4)吴恩达网易云课程:https://study.163.com/my#/smarts
(5)刘洪普老师博客:https://liuii.github.io/
(6)某同学的笔记:http://biranda.top/archives/page/2/

标签:loss,基础教程,grad,0.01,item,PyTorch,反向,计算,data
来源: https://blog.csdn.net/qq_35812205/article/details/120814447

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有