src/transformers/models/albert/modeling_albert.py/AlbertSOPHead/__init__
class AlbertSOPHead:    def __init__(self, config: AlbertConfig):
        super().__init__()

        self.dropout = nn.Dropout(config.classifier_dropout_prob)
        self.classifier = nn.Linear(config.hidden_size, config.num_labels)