src/transformers/models/albert/modeling_albert.py/AlbertForPreTraining/__init__
class AlbertForPreTraining:    def __init__(self, config):
        super().__init__(config)

        self.albert = AlbertModel(config)
        self.predictions = AlbertMLMHead(config)
        self.sop_classifier = AlbertSOPHead(config)

        self.init_weights()