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