class torchelie.models.ProjectionDiscr(in_channels: int, num_classes: int)

Experimental: A classification head for conditional GANs discriminators using a projection discriminator .

  • feat_extractor (nn.Module) – a feature extraction model

  • feature_size (int) – the number of features in the last layer of the feature extractor

  • num_classes (int) – the number of output classes


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

training: bool