Pix2PixGenerator

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

UNet generator from Pix2Pix. Dropout layers have been substitued with Noise injections from StyleGAN2.

Parameters

arch (List[int]) – the number of channel for each depth level of the UNet.

set_padding_mode(mode: str) → torchelie.models.pix2pix.Pix2PixGenerator

to_equal_lr() → torchelie.models.pix2pix.Pix2PixGenerator

to_instance_norm(affine: bool = True) → torchelie.models.pix2pix.Pix2PixGenerator

Pix2Pix sometimes uses batch size 1, similar to instance norm.

training: bool