admin 管理员组

文章数量: 1184232

在进行Gan的训练过程中，经常会遇到这个问题：RuntimeError: Trying to backward through the graph a second time, but the saved intermediate results have already been freed

这个错误的描述就是，你在进行方向传播的时候，缓存提前释放了，很多解决办法都是说在loss.backward() 改成loss.backward(retain_graph=True)，实际上，大部分的代码问题不出在这里，而且保留计算图可能会使缓存快速积累，导致显存爆了，其实真正原因是由于判别器和生产器公用了相同的变量，只要在第一次用变量时候加上detach()就可以了。

#### Update Discriminator ###
real_preds = netD(real_gt)
fake_preds = netD(fake_coarse)


#### Update Generator #####
real_preds = netD(real_gt)
fake_preds = netD(fake_coarse)

改成：

#### Update Discriminator ###
real_preds = netD(real_gt.detach())
fake_preds = netD(fake_coarse.detach())


#### Update Generator #####
real_preds = netD(real_gt)
fake_preds = netD(fake_coarse)

查找方法不易，感谢支持

本文标签： 报错 pytorch RuntimeError Time Graph

版权声明：本文标题：pytorch报错：RuntimeError: Trying to backward through the graph a second time 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1755028187a3062480.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。