examples/research_projects/movement-pruning/emmental/modeling_bert_masked.py/BertAttention/__init__
class BertAttention:    def __init__(self, config):
        super().__init__()
        self.self = BertSelfAttention(config)
        self.output = BertSelfOutput(config)
        self.pruned_heads = set()