TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

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.