pytorch with torch.no_grad

本文详细介绍了Python中的with语句及其在资源管理中的作用,然后重点讲解了PyTorch中torch.no_grad()的功能。torch.no_grad()用于在计算过程中禁止自动求导,从而节省内存。当在with torch.no_grad()块内进行运算时,即使原本要求梯度的张量也会在该范围内不记录梯度,这对于大规模模型或在评估阶段特别有用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

【pytorch系列】 with torch.no_grad():用法详解_大黑山修道的博客-优快云博客_torch.no_grad():在pytorch写的网络中,with torch.no_grad():非常常见。首先,关于python中的with:with 语句适用于对资源进行访问的场合,确保不管使用过程中是否发生异常都会执行必要的“清理”操作,释放资源,比如文件使用后自动关闭/线程中锁的自动获取和释放等。例如:file = open("1.txt")data = file.read()file.close()存在问题如下:(1)文件读取发生异常,但没有进行任何处理;(2)可能忘记关闭文件句柄;改进:try: https://blog.youkuaiyun.com/sazass/article/details/116668755

首先,关于python中的with

with 语句适用于对资源进行访问的场合,确保不管使用过程中是否发生异常都会执行必要的“清理”操作,释放资源,比如文件使用后自动关闭/线程中锁的自动获取和释放等。

关于 torch.no_grad()
首先从requires_grad讲起:

requires_grad
在pytorch中,tensor有一个requires_grad参数,如果设置为True,则反向传播时,该tensor就会自动求导。tensor的requires_grad的属性默认为False,若一个节点(叶子变量:自己创建的tensor)requires_grad被设置为True,那么所有依赖它的节点requires_grad都为True(即使其他相依赖的tensor的requires_grad = False)

当requires_grad设置为False时,反向传播时就不会自动求导了,因此大大节约了显存或者说内存。

with torch.no_grad的作用
在该模块下,所有计算得出的tensor的requires_grad都自动设置为False。

即使一个tensor(命名为x)的requires_grad = True,在with torch.no_grad计算,由x得到的新tensor(命名为w-标量)requires_grad也为False,且grad_fn也为None,即不会对w求导。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值