ProjectionDiscr¶

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

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

Parameters

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

Warning

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