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

        self.embeddings = BertEmbeddings(config)
        self.embeddings.requires_grad_(requires_grad=False)
        self.encoder = BertEncoder(config)
        self.pooler = BertPooler(config)

        self.init_weights()