美文网首页
torch.no_grad

torch.no_grad

作者: He_Yu | 来源:发表于2019-10-03 15:39 被阅读0次

    PyTorch里的requires_grad、volatile及no_grad

    requires_grad=True 要求计算梯度

    requires_grad=False 不要求计算梯度

    with torch.no_grad()或者@torch.no_grad()中的数据不需要计算梯度,也不会进行反向传播

    torch.no_grad()是新版本pytorch中volatile的替代)

    相关文章

      网友评论

          本文标题:torch.no_grad

          本文链接:https://www.haomeiwen.com/subject/apvdpctx.html