examples/research_projects/movement-pruning/emmental/modeling_bert_masked.py/MaskedBertForQuestionAnswering/__init__
class MaskedBertForQuestionAnswering:    def __init__(self, config):
        super().__init__(config)
        self.num_labels = config.num_labels

        self.bert = MaskedBertModel(config)
        self.qa_outputs = nn.Linear(config.hidden_size, config.num_labels)

        self.init_weights()