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

        self.bert = BertModelWithPabee(config)
        self.dropout = nn.Dropout(config.hidden_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()