在执行PyTorch代码时，遇到一个问题：由于一个变量在被用于梯度计算时进行了inplace操作，导致出现了'variables needed for gradient computation has been modified by an inplace operation'的错误。具体来说，问题出在self.conf加了cuda()之后，尝试对self.conf_mask进行操作。为解决此问题，应该避免使用inplace操作，如将`self.conf_mask[...] = 0`改为`self.conf_mask.cpu()[...] = 0`。
one of the variables needed for gradient computation has been modified by an
self.conf加了cuda()之后就成这样了。
这个变量不求导
self.conf_mask = torch.zeros(batch_size,self.num_anchors,g_dim, g_dim, requires_grad=False).cuda()
错误代码：
self.conf_mask[...] = 0
正确做法：
self.conf_mask.cpu()[...] = 0
2953
292
2799
1576
348
746
1063
756
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
超级会员免费看
扫一扫
