examples/research_projects/bert-loses-patience/pabee/modeling_pabee_albert.py/AlbertForSequenceClassificationWithPabee/__init__
class AlbertForSequenceClassificationWithPabee:    def __init__(self, config):
        super().__init__(config)
        self.num_labels = config.num_labels

        self.albert = AlbertModelWithPabee(config)
        self.dropout = nn.Dropout(config.classifier_dropout_prob)
        self.classifiers = nn.ModuleList(
            [nn.Linear(config.hidden_size, self.config.num_labels) for _ in range(config.num_hidden_layers)]
        )

        self.init_weights()