tensorflow - 训练 GAN 时，是否应在禁用训练时在鉴别器中禁用 dropout？

Question

我正在 keras 中执行基本的 GAN 实现。训练是分阶段进行的，首先单独训练判别器，然后将生成器作为组合模型(生成器加判别器)的一部分进行训练，判别器禁用训练。如果鉴别器中有 dropout，在我看来，在通过组合模型训练生成器时应该禁用它。这是真的？如果是这样，如何在keras中完成？我知道设置 trainable=false 不会禁用 dropout。

Answer 1

你是对的，在训练判别器或任何测试阶段时，应该为生成器禁用 dropout。好消息是 keras 默认情况下会这样做 link .

所以看看您的场景，您可以使用 trainable 标志调用生成器的预测函数，并使用它作为输入来训练鉴别器。

希望这能回答您的问题。