PyTorch 多个网络反向传播时出现梯度计算错误

torch具体报错内容如下

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

我现在正在做的算法是关于强化学习MADDPG,需要用到两个网络actor和critic

经过调试之后发现,我在计算出critic loss之后并没有进行critic网络的反向传播,而是开始计算actor loss。两个网络的loss都计算完毕后才开始进行反向传播就会出现上述报错。

解决方法就是将两个网络的loss和backward都单独封装计算即可

相关推荐

  1. PyTorch 网络反向传播出现梯度计算错误

    2024-03-14 10:28:05       21 阅读
  2. PyTorch】前向传播反向传播计算

    2024-03-14 10:28:05       33 阅读
  3. Pytorch 反向传播 计算图被修改的报错

    2024-03-14 10:28:05       40 阅读

最近更新

  1. 根据关键词query获取google_img(api方式)

    2024-03-14 10:28:05       0 阅读
  2. redis中的事务和mysql中的事务有什么区别?

    2024-03-14 10:28:05       1 阅读
  3. C# 构造函数依赖注入 使用out向外传递参数

    2024-03-14 10:28:05       1 阅读
  4. 信息时代,呼唤新的哲学

    2024-03-14 10:28:05       1 阅读
  5. 【数据基础】— B树

    2024-03-14 10:28:05       1 阅读
  6. Vue 路由传参 query方法 bug 记录

    2024-03-14 10:28:05       1 阅读
  7. 翻页 上一页/下一页

    2024-03-14 10:28:05       1 阅读

热门阅读

  1. 计算机网络——分组延时、丢失和吞吐量

    2024-03-14 10:28:05       20 阅读
  2. Oracle Linux 8.9 安装 Postman

    2024-03-14 10:28:05       20 阅读
  3. 【OpenCV实战】基于OpenCV的传统手势识别流程详解

    2024-03-14 10:28:05       23 阅读
  4. 跨域的几种方式?(详解)

    2024-03-14 10:28:05       25 阅读
  5. ChatGLM3 源码解析(五)

    2024-03-14 10:28:05       18 阅读
  6. Apache Spark 的基本概念和在大数据分析中的应用

    2024-03-14 10:28:05       25 阅读
  7. 项目使用jdk17启动报错

    2024-03-14 10:28:05       25 阅读
  8. 原型和原型链的区别,__proto__和prototype的区别

    2024-03-14 10:28:05       21 阅读
  9. Go语言的自给自足:编译自身的神奇之旅

    2024-03-14 10:28:05       27 阅读
  10. 【Docker】Tensorflow 容器化部署

    2024-03-14 10:28:05       22 阅读
  11. 预取和缓存替换介绍--自用

    2024-03-14 10:28:05       20 阅读
  12. 【WEEK2】学习目标及总结【SpringMVC】【中文版】

    2024-03-14 10:28:05       23 阅读
  13. Spring MVC InternalResourceViewResolver原理解析

    2024-03-14 10:28:05       23 阅读
  14. Goland运行go语言基础篇

    2024-03-14 10:28:05       23 阅读