with torch.no_grad() 和 @torch.no_grad()


Pytorch中with torch.no_grad()或@torch.no_grad() 用法

https://www.cnblogs.com/douzujun/p/13364116.html

  • 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():
	...

原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/289931.html

(0)
上一篇 2022年9月17日
下一篇 2022年9月17日

相关推荐

发表回复

登录后才能评论