mirror of
https://github.com/TencentARC/GFPGAN.git
synced 2026-02-15 05:44:38 +00:00
GAN训练 #59
Reference in New Issue
Block a user
Delete Branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
Originally created by @SimKarras on GitHub (Aug 25, 2021).
您好,有一个问题困扰我。在GAN的训练中,参考https://github.com/rosinality/stylegan2-pytorch/blob/master/train.py
在训练Discriminator时,rosinality将Generator的梯度更新关闭:
同样,训练Generator时,也会将Discriminator的梯度更新关闭:
我只在您的代码中找到了对Discriminator进行梯度控制,没有对Generator的梯度调节:
&
1、这是不是意味着Generator始终会得到梯度更新(哪怕是在训练Discriminator时)?如果是这样,是否等价于每份数据都会在Generator前降传播两次呢?
2、如果Generator的梯度更新也会受到调节,请问这是在哪个位置实现的呢?
@xinntao commented on GitHub (Aug 29, 2021):
当我们在更新 D的时候, G的输出进行了detach操作,所以是没有梯度传递到G的

@SimKarras commented on GitHub (Aug 30, 2021):
我明白了,十分感谢您的解惑!