GANEstimator


Analytics-Zoo provides a GANEstimator to support training GAN like models. Currently we support standard unconditional/conditional GAN and other GAN types will be supported in the future.


GANEstimator

Python

GANEstimator(generator_fn,
             discriminator_fn,
             generator_loss_fn,
             discriminator_loss_fn,
             generator_optimizer,
             discriminator_optimizer,
             generator_steps=1,
             discriminator_steps=1,
             model_dir=None,
    )

Arguments

train

estimator.train(input_fn=input_fn, end_trigger=MaxIteration(100))

Arguments