RUAS代码debug

  • 最近训练Retinex-inspired Unrolling with Cooperative Prior Architecture Search for Low-light Image Enhancement的代码(https://github.com/KarelZhang/RUAS)的时候报错说梯度回传有问题,有变量被修改了,仔细一看源码,问题出在这里:
  • 在这里插入图片描述
  • model.py文件中,226行是报错位置,也就是说denoise_loss backward的时候,与denoise_loss有梯度回传关系的张量或者网络参数被修改了,导致torch检查的时候,denoise loss来自version 50的张量,回传到某个张量的时候,这个张量的版本已经变成version 51了,其实很好理解,因为每50次才会运行到一次226行,所以问题出在前面enhancement_loss的部分
  • 问题出在220行,220行先把enhance_net的参数更新掉了,而到226行梯度回传到enhance_net,此时enhance_net的版本已经不是产生denoise_loss的版本了。要改也很简单,因为从算法上,enhance_net和denoise_net其实并不是联合训练关系,梯度完全没有必要回传到enhance_net,只需要掐断就好了。那么梯度是怎么传到enhance_net的呢,其实是192行,enhance_net生成的u_list进一步传到了denoise_net,导致回传的时候通过u_list回传到了enhance_net,那么只需要在这里掐断就好了,加个detach:
    在这里插入图片描述
  • 同时225行算loss的时候也用到了u_list,也掐断:
    在这里插入图片描述
  • 改完就能正常训练了。注意这个改法是建立在明确知道denoise_net和enhance_net没有任何联合训练关系的前提,否则不建议随意掐断梯度。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值