Pytorch中with torch.no_grad()或@torch.no_grad() 用法
作者:互联网
参考Pytorch中with torch.no_grad()或@torch.no_grad() 用法 - 云+社区 - 腾讯云
-
requires_grad=True
要求计算梯度 -
requires_grad=False
不要求计算梯度 -
with torch.no_grad()
或者@torch.no_grad()
中的数据不需要计算梯度,也不会进行反向传播
model.eval() # 测试模式
with torch.no_grad():
pass
@torch.no_grad()
def eval():
...
标签:no,梯度,torch,Pytorch,eval,grad 来源: https://blog.csdn.net/weixin_36670529/article/details/112452384