Pix2PixHDGlobalGenerator

class torchelie.models.Pix2PixHDGlobalGenerator(arch: List[str])

Residual generator used in Pix2PixHD .

arch is a list of strings representing blocks.

For example, this creates a first conv with 32 output channels, 3 downsampling stride 2 convs that double the number of channels, 5 residual blocks, 3 upsampling convs halving the number of channels, and a final conv that converts back to RGB.

:code:``` Pix2PixHDGlobalGenerator([‘32’, ‘d128’, ‘d512’, ‘d512’, ‘R512’, ‘R512’,

‘R512’, ‘R512’, ‘R512’, ‘u512’, ‘u512’, ‘u128’])

```

leaky() → torchelie.models.pix2pixhd.Pix2PixHDGlobalGenerator

to_equal_lr(leak=0.0) → torchelie.models.pix2pixhd.Pix2PixHDGlobalGenerator

to_standard_arch()

to_unet() → torchelie.models.pix2pixhd.Pix2PixHDGlobalGenerator

training: bool