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¶
-