def METHOD_NAME(self, args):
    from fairseq import models
    model = models.METHOD_NAME(args, self)
    model.register_classification_head(
        'sentence_classification_head',
        num_classes=self.args.num_classes,
    )
    return model