TensorFlow 2.0 RC is available Learn more

tf.contrib.gan.get_sequential_train_hooks

View source on GitHub

Returns a hooks function for sequential GAN training.

tf.contrib.gan.get_sequential_train_hooks(train_steps=namedtuples.GANTrainSteps(1, 1))

Args:

  • train_steps: A GANTrainSteps tuple that determines how many generator and discriminator training steps to take.

Returns:

A function that takes a GANTrainOps tuple and returns a list of hooks.