def METHOD_NAME(self):
    head_cfg = super().METHOD_NAME()
    if hasattr(head_cfg, 'classifier_dropout'):
        head_cfg['classifier_dropout'] = (
            head_cfg.classifier_dropout if head_cfg['classifier_dropout']
            is not None else head_cfg.hidden_dropout_prob)
    else:
        head_cfg['classifier_dropout'] = head_cfg.hidden_dropout_prob
    head_cfg['num_labels'] = self.config.num_labels
    return head_cfg