src.models.classification_head

Classes

ClassificationHead(input_size, num_classes)

Standard classification head for MLP and EMD loss models.