src/transformers/models/ernie/modeling_ernie.py/ErnieOnlyNSPHead/forward
class ErnieOnlyNSPHead:    def forward(self, pooled_output):
        seq_relationship_score = self.seq_relationship(pooled_output)
        return seq_relationship_score