examples/research_projects/deebert/src/modeling_highway_bert.py/BertHighway/__init__
class BertHighway:    def __init__(self, config):
        super().__init__()
        self.pooler = BertPooler(config)
        self.dropout = nn.Dropout(config.hidden_dropout_prob)
        self.classifier = nn.Linear(config.hidden_size, config.num_labels)