RuntimeError: one of the variables needed forgradient computation has been modifiedby an inplace ope

遇到一个报错如下:

RuntimeError: one of the variables needed forgradient computation has been modifiedby an inplace operatijon.
在这里插入图片描述

解决方法:
只需找个地方,比如在你怀疑有问题的地方,加上他说的这句话:

torch.autograd.seg_detect_anomaly(True)

再次运行,就会显示更详细的信息和更精确的定位。

在相应位置修改即可。比如我的是:
第527行对用v1_0的修改是in place操作
RuntimeError: one of the variables needed forgradient computation has been modifiedby an inplace ope_第1张图片

改成换一个变量名就行:
RuntimeError: one of the variables needed forgradient computation has been modifiedby an inplace ope_第2张图片

你可能感兴趣的:(个人笔记,AI,Python,深度学习,pytorch,cuda)