关于pytorch实现GAN中训练过程detach函数的思考
问题描述在GAN的训练过程中,训练G(生成器)时需要将D(判别器)固定。训练D时需要将G固定。在网上找的训练程序中有detach函数。网上查阅,detach函数可以切断反向梯度传播。先放代码if ii % opt.d_every == 0: # 训练判别器 optimizer_d.zero_grad() ## 尽可能的把真图片判别为正确 output = netd(real_img) error_d_real = criterion(output, true_l
原创
2020-08-01 21:21:32 ·
888 阅读 ·
2 评论