适用于`zero_gradients`的PyTorch 1.9命令等效命令
适用于`zero_gradients`的PyTorch 1.9命令等效命令
提问于 2021-07-22 21:20:25
这个错误是在我的PyTorch版本从1.8升级到1.9.0之后出现的。
使用此行时：
from torch.autograd.gradcheck import zero_gradients，我收到这个错误消息：ImportError: cannot import name 'zero_gradients' from 'torch.autograd.gradcheck'
命令：
zero_gradients(im)
是使用的。
PyTorch 1.9.0中的新命令相当于什么？
回答 1
Stack Overflow用户
回答已采纳
发布于 2021-07-22 21:45:45
在pytorch中没有名为zero_gradients()的函数。最接近的相似名称是zero_grad()。相反，它是您在auto-attack/autoattack/other_utils.py 中共享的代码库中定义的函数。
def zero_gradients(x):
if isinstance(x, torch.Tensor):
if x.grad is not None:
x.grad.detach_()
x.grad.zero_()
elif isinstance(x, container_abcs.Iterable):
for elem in x:
zero_gradients(elem)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/68485574
复制相关文章
相似问题
腾讯云开发者
