|View source on GitHub|
An InfoGANModel contains all the pieces needed for InfoGAN training.
See https://arxiv.org/abs/1606.03657 for more details.
structured_generator_inputs: A list of Tensors representing the random noise that must have high mutual information with the generator output. List length should match
predicted_distributions: A list of
tfp.distributions.Distributions. Predicted by the recognizer, and used to evaluate the likelihood of the structured noise. List length should match
discriminator_and_aux_fn: The original discriminator function that returns a tuple of (logits,