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

Summary¶

Classes:

NoPatchDiscriminator

Reference¶

class NoPatchDiscriminator(input_nc: int, norm_layer: torch.nn.modules.module.Module = <class 'torch.nn.modules.batchnorm.BatchNorm2d'>, n_layers_d: int = 4, use_sigmoid: bool = True)[source]¶

Bases: torch.nn.modules.module.Module

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

Forwarding through network and avg pooling.

Parameters: x (torch.Tensor) – the input tensor

training: bool¶