GAN使用填坑过程

GAN使用填坑过程:

1、RuntimeError: Trying to backward through the graph a second time, but the buffers have already been freed. Specify retain_graph=True when calling backward the first time.

原因:GAN的训练过程中,存在两个loss的梯度更新,loss.backward(),如果没有交替更新判别网络和生成网络,嵌套的GAN网络训练时就会出现该问题。在默认情况下,网络在反向传播中不允许多个backward()。

解决方法:需要在第一个backward设置retain_variables=True,即loss1.backward(retain_variables=True). 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值