zoo.tfpark.gan package

Submodules

zoo.tfpark.gan.common module

class zoo.tfpark.gan.common.GanOptimMethod(d_optim, g_optim, g_param_size, d_steps=1, g_steps=1)

Bases: bigdl.optim.optimizer.OptimMethod

zoo.tfpark.gan.gan_estimator module

class zoo.tfpark.gan.gan_estimator.GANEstimator(generator_fn, discriminator_fn, generator_loss_fn, discriminator_loss_fn, generator_optimizer, discriminator_optimizer, generator_steps=1, discriminator_steps=1, model_dir=None, session_config=None)

Bases: object

train(input_fn, end_trigger)

Module contents