- 博客(1)
- 收藏
- 关注
原创 一行代码解决one of the variables needed for gradient computation has been modified by an inplace operation
如果是在对抗中,“因为算G的loss的计算图是包含D的,但是你在G的backward之前更新了D的值,这时候再去计算就不是和forward时候对应的梯度了”(出自虎扑),解释一下就是算D_loss时候已经改变了fake的inplace了,再去算G_loss时候再用fake这个变量就会报错了,解决方案就是在算G_loss前,clone一下这个fake变量就可以,或者再通过网络输出一遍。
2023-12-22 11:57:29 926 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人