src/transformers/models/roformer/modeling_roformer.py/RoFormerClassificationHead/__init__
class RoFormerClassificationHead:    def __init__(self, config):
        super().__init__()
        self.dense = nn.Linear(config.hidden_size, config.hidden_size)
        self.dropout = nn.Dropout(config.hidden_dropout_prob)
        self.out_proj = nn.Linear(config.hidden_size, config.num_labels)

        self.config = config