simulation.utils.machine_learning.cycle_gan.models.n_layer_discriminator module¶

Summary¶

Classes:

NLayerDiscriminator

Defines a PatchGAN discriminator.

Reference¶

class NLayerDiscriminator(input_nc: int, ndf: int = 64, n_layers: int = 3, norm_layer: torch.nn.modules.module.Module = <class 'torch.nn.modules.batchnorm.BatchNorm2d'>, use_sigmoid: bool = True, is_quadratic: bool = True)[source]¶

Bases: torch.nn.modules.module.Module

Defines a PatchGAN discriminator.

forward(input: torch.Tensor) → torch.Tensor[source]¶

Standard forward.

Parameters: input (Tensor) – the input tensor

training: bool¶