报错解决:one of the variables needed for gradient computation has been modified by an inplace operation

写完了训练的代码,运行了以后发现了这么一个错:

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.FloatTensor [544, 768]], which is output 0 of ViewBackward, 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).

大概意思就是说在计算梯度的时候检查出某个Variable有被一个 inplace operation 修改。

照着提示信息,设置:torch.autograd.set_detect_anomaly = True,再运行一下,得到下面更详细的输出:

Traceback (most recent call last):
  File "E:\xxx\main\main.py", line 71, in <module>
    main(args)
  File "E:\xxx\main\main.py", line 57, in main
    train(config)
  File "E:\xxx\train\train.py", line 114, in train
    total_loss.backward()
  File "D:\Anaconda3\envs\nlp\lib\site-packages\torch\tensor.py", line 245, in backward
    torch.autograd.backward(self, gradient, retain_graph, create_graph, inputs=inputs)
  File "D:\Anaconda3\envs\nlp\lib\site-packages\torch\autograd\__init__.py", line 145, in backward
    Variable._execution_engine.run_backward(
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.FloatTensor [544, 768]], which is output 0 of ViewBackward, 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).

可以看到是反向传播的过程中出错的。


那么什么是 inplace operation 呢?

根据pytorch论坛的问答帖子:

inplace operation 就是直接对tensor的内容进行修改,而没有使用复制的副本 (An in-place operation is an operation that changes directly the content of a given Tensor without making a copy)。

  • 在pytorch中, inplace operation 可以是一些 .add_().scatter_() 导致的。对于.add_()方法,是直接在tensor上进行修改的,可以把x.add_(y)改成x = x + y。如果需要复制一个副本话,参照第二个帖子的方法,可以使用.clone()方法。

  • 在python中, inplace operation 可以是一些 +=*= 导致的。比如 x += y,需要改成 x = x +y

### 解决 PyTorch 中由原地操作引发的 RuntimeError 当遇到 `RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation` 错误时,这通常意味着某些张量在反向传播过程中被修改了其数据,而这些修改发生在计算图中的节点之后。这种情况下,PyTorch 无法正确追踪梯度的变化。 为了处理这种情况,可以采取以下几个措施: #### 启用异常检测 启用自动求导模块中的异常检测功能可以帮助定位具体在哪一步发生了问题。通过设置 `torch.autograd.set_detect_anomaly(True)` 可以让程序在每次前向传递后检查是否存在不合法的操作,并抛出更详细的错误信息以便调试[^1]。 ```python import torch # 开启anomaly detection模式来帮助查找问题所在 torch.autograd.set_detect_anomaly(True) # 继续执行训练循环... ``` #### 避免使用原地运算符 许多 PyTorch 的函数都有对应的原地版本(带有下划线 `_`),比如 `.add_()` 或者 `.relu_()`. 这些方法会直接改变输入张量的内容而不是创建新的对象返回。为了避免破坏计算图结构,在构建模型或者编写自定义层时应尽量避免使用这类原地操作[^2]。 例如,如果原本有如下代码片段: ```python output = F.relu(output, inplace=True) ``` 应该改为非inplace的形式: ```python output = F.relu(output) ``` #### 使用 detach 方法分离不需要跟踪的历史记录 对于那些确实需要做原地更新但是又不想影响到整个计算图的情况,可以通过调用 `.detach()` 来切断当前张量与其历史之间的联系,从而允许对其进行安全的原地更改而不干扰后续的梯度计算过程[^3]。 ```python detached_output = output.detach() detached_output.add_(some_value) # 对 detached_output 执行原地加法不会影响原始 tensor 的 history ``` #### 修改网络架构设计 有时,特定类型的神经元激活函数可能会更容易触发此类错误,特别是像 ReLU 和 Tanh 这样的饱和型激活函数。考虑调整使用的激活函数种类或是重新审视整体网络的设计逻辑是否合理[^4]。 综上所述,针对此 Runtime Error 主要策略包括开启 anomaly detection 辅助排查、禁用所有可能引起冲突的原地操作以及适当运用 .detach() 技巧等手段相结合的方式来进行修复工作。
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值