其他分享
首页 > 其他分享> > Pytorch中with torch.no_grad()或@torch.no_grad() 用法

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

作者:互联网

参考Pytorch中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