问从zero_gradients中模仿torch.autograd.gradcheck
EN

Stack Overflow用户

提问于 2021-07-17 10:24:06

回答 1查看 4K关注 0票数 4

我想复制代码这里，在Google中运行时会得到以下错误？

(/usr/local/lib/python3.7/dist-packages/torch/autograd/gradcheck.py)：无法从“torch.autograd.gradcheck”torch.autograd.gradcheck导入名称“zero_gradients”

有人能帮我解决这个问题吗？

发布于 2021-07-17 14:41:42

这似乎是在使用一个非常旧的PyTorch版本，这个函数本身不再可用了。但是，如果您查看此承诺，您将看到zero_gradients的实现。它所做的只是将输入的梯度为零：

def zero_gradients(i):
    for t in iter_gradients(i):
        t.zero_()

那么zero_gradients(x)应该与x.zero_grad()相同，这是当前的API，假设x是nn.Module！

或者只是：

if x.grad is not None:
    x.grad.zero_()

票数 3

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/68419612

复制

相似问题

问从zero_gradients中模仿torch.autograd.gradcheckEN