neodroidvision.regression.generative.adversarial.sagan

Created on 19-09-2021

Classes

ConditionalBatchNorm2d(num_features, num_classes)

https://github.com/pytorch/pytorch/issues/8985#issuecomment-405080775

Discriminator(d_conv_dim, num_classes)

Discriminator.

DiscriminatorBlock(in_channels, out_channels)

description

DiscriminatorOptBlock(in_channels, out_channels)

description

GenBlock(in_channels, out_channels, num_classes)

description

Generator(z_dim, g_conv_dim, num_classes)

Generator.