src/transformers/models/efficientnet/modeling_efficientnet.py/EfficientNetForImageClassification/__init__
class EfficientNetForImageClassification:    def __init__(self, config):
        super().__init__(config)
        self.num_labels = config.num_labels
        self.config = config
        self.efficientnet = EfficientNetModel(config)
        # Classifier head
        self.dropout = nn.Dropout(p=config.dropout_rate)
        self.classifier = nn.Linear(config.hidden_dim, self.num_labels) if self.num_labels > 0 else nn.Identity()
        self.classifier_act = nn.Softmax(dim=1)

        # Initialize weights and apply final processing
        self.post_init()