Pytorch中with torch.no_grad()或@torch.no_grad() 用法
Pytorch中with torch.no_grad()或@torch.no_grad() 用法
狼啸风云
修改于 2022-09-02 19:50:47
修改于 2022-09-02 19:50:47
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():
...本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2021-02-09 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读