examples/research_projects/movement-pruning/emmental/modeling_bert_masked.py/MaskedBertModel/set_input_embeddings
class MaskedBertModel:    def set_input_embeddings(self, value):
        self.embeddings.word_embeddings = value