TensorFlow 2.0 Beta is available Learn more

Module: tf.contrib.gan.estimator

TF-GAN estimator module.

Defined in contrib/gan/python/estimator/__init__.py.

GANEstimator provides all the infrastructure support of a TensorFlow Estimator with the feature support of TF-GAN.

Modules

gan_estimator module: tf.Learn components for GANEstimator.

head module: tf.Learn components for GANEstimator's loss.

latent_gan_estimator module: tf.Learn components for Train Input Estimator.

stargan_estimator module: tf.Learn components for GANEstimator.

tpu_gan_estimator module: tf.Learn components for TPUGANEstimator.

Classes

class GANEstimator: An estimator for Generative Adversarial Networks (GANs).

class GANHead: Head for a GAN.

class StarGANEstimator: An estimator for Generative Adversarial Networks (GANs).

class SummaryType

class TPUGANEstimator: An estimator for Generative Adversarial Networks (GANs) on TPU.

Functions

gan_head(...): Creates a GANHead. (deprecated)

get_latent_gan_estimator(...): Gets an estimator that passes gradients to the input.