models package

Submodules

models.brain module

models.brain.critic_fn(gan_type)

Return critic model_fn for gan_type

Parameters:gan_type (str) – Either “COND” or “UNCOND” Returns:model_fn for GAN critic Return type:critic_fn

models.brain.generator_fn(gan_type)

Return generator model_fn for gan_type

Parameters:gan_type (str) – Either “COND” or “UNCOND” Returns:model_fn for GAN generator Return type:generator_fn

models.mnist module

models.mnist.critic_fn(gan_type)

Return critic model_fn for gan_type

Parameters:gan_type (str) – Either “COND” or “UNCOND” Returns:model_fn for GAN critic Return type:critic_fn

models.mnist.generator_fn(gan_type)

Return generator model_fn for gan_type

Parameters:gan_type (str) – Either “COND” or “UNCOND” Returns:model_fn for GAN generator Return type:generator_fn