StyleGAN2Discriminator

class torchelie.models.StyleGAN2Discriminator(input_sz, max_ch: int = 512, ch_mul: int = 1)

Experimental: Build the discriminator for StyleGAN2

Parameters

• input_sz (int) – image size

• max_ch (int) – maximum number of channels (default: 512)

• ch_mul (float) – multiply the number of channels on each layer by this value (default, 1.)

Warning

StyleGAN2Discriminator() is experimental, and may change or be deleted soon if not already broken