GAN_CIFAR10_jax module

class GAN_CIFAR10_jax.Discriminator(parent: ~flax.linen.module.Module | ~flax.core.scope.Scope | ~flax.linen.module._Sentinel | None = <flax.linen.module._Sentinel object>, name: str | None = None)

Bases: Module

Discriminator class for an unconditional GAN model. Applies several convolutional layers followed by LeakyReLU activations to classify the input.

__call__()

The forward pass through the network.

name: str | None = None

parent: Module | Scope | _Sentinel | None = None

scope: Scope | None = None

class GAN_CIFAR10_jax.Discriminator_cond(parent: ~flax.linen.module.Module | ~flax.core.scope.Scope | ~flax.linen.module._Sentinel | None = <flax.linen.module._Sentinel object>, name: str | None = None)

Bases: Module

Conditional Discriminator class for a GAN model. Takes both images and labels as input and discriminates between real and fake images conditioned on the labels.

__call__()

The forward pass through the network.

name: str | None = None

parent: Module | Scope | _Sentinel | None = None

scope: Scope | None = None

class GAN_CIFAR10_jax.Generator(parent: ~flax.linen.module.Module | ~flax.core.scope.Scope | ~flax.linen.module._Sentinel | None = <flax.linen.module._Sentinel object>, name: str | None = None)

Bases: Module

Generator class for an unconditional GAN model. Takes a latent code and generates images using deconvolutional layers.

__call__()

The forward pass through the network.

name: str | None = None

parent: Module | Scope | _Sentinel | None = None

scope: Scope | None = None

class GAN_CIFAR10_jax.Generator_cond(parent: ~flax.linen.module.Module | ~flax.core.scope.Scope | ~flax.linen.module._Sentinel | None = <flax.linen.module._Sentinel object>, name: str | None = None)

Bases: Module

Conditional Generator class for a GAN model. Takes latent codes and labels to generate conditional images.

__call__()

The forward pass through the network.

name: str | None = None

parent: Module | Scope | _Sentinel | None = None

scope: Scope | None = None