PatchDiscriminator

class torchelie.models.PatchDiscriminator(arch: List[int])

remove_batchnorm() → torchelie.models.patchgan.PatchDiscriminator

set_input_specs(in_channels: int) → torchelie.models.patchgan.PatchDiscriminator

set_kernel_size(kernel_size: int) → torchelie.models.patchgan.PatchDiscriminator

to_avg_pool() → torchelie.models.patchgan.PatchDiscriminator

to_binomial_downsampling() → torchelie.models.patchgan.PatchDiscriminator

to_equal_lr(leak=0.2) → torchelie.models.patchgan.PatchDiscriminator

to_instance_norm(affine: bool = True) → torchelie.models.patchgan.PatchDiscriminator

Pix2PixHD uses instancenorm rather than batchnorm

training: bool