关于GAN训练过程中的报错:one of the variables needed for gradient computation has been modified by an inplace
首先说明,按照我目前的查询,这可能是全网唯一公开的正确解决方法,所以一定要看下去1、错误发生情景在github和百度上搜索gan示例代码的时候,通常会得到下面这种代码:先更新辨别器,再更新生成器。 netD.zero_grad() #optimizerD.step() real_out = netD(real_img).mean() fake_out = netD(fake_img).mean()
原创
2022-02-24 13:49:30 ·
5552 阅读 ·
15 评论