RuntimeError: one of the variables needed for gradient computation has been modified by an inplace o

悄悄地努力

已于 2024-04-12 17:42:44 修改

阅读量965

点赞数 3

分类专栏： bug 解决文章标签：计算机视觉 python

于 2024-04-12 17:40:27 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_46034990/article/details/137688979

版权

文章讲述了在使用PyTorch时遇到的RuntimeError，原因是模型定义中的一个SEBlock使用了inplace=True导致变量版本不匹配。解决方案是移除inplace属性并提供了一个修复后的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [2, 256, 1, 1]], which is output 0 of ReluBackward0, is at version 1; expected version 0 instead. Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).

问题分析

在原模型定义中增加了一个 SE 块，有以下代码：

self.headpose_feature = nn.Sequential(
	nn.ReLU(inplace=True),
	SELaye

最低0.47元/天解锁文章