使用带有 GAN 的 tensorflow 输入管道

Question

我想用 tensorflow 构建一个条件 GAN，并使用输入管道来加载我的数据集。问题是，在每次迭代中，我想使用相同的数据批次来训练生成模型和判别模型，但是由于它们的训练操作符是在不同的运行中获取的，因此它们将接收到不同批次的数据。有什么解决方案或者我应该使用 feed_dict 吗？

Answer 1

使用相同数据的一种方法是在生成器和判别器训练操作上使用tf.group，以便联合训练它们，并在优化器上设置use_locking=True 以防止病态竞争条件。请注意，由于 TensorFlow 运行时无法保证始终首先训练生成器或判别器，因此仍然存在一些随机性。

这个想法已经在 TensorFlow 的 TFGAN 库get_joint_train_hooks 中实现，尽管它使用钩子而不是对训练操作进行分组（“联合”指的是鉴别器和生成器是联合训练的，而不是顺序训练的）。