src/transformers/models/megatron_bert/modeling_megatron_bert.py/MegatronBertOnlyNSPHead/forward
class MegatronBertOnlyNSPHead:    def forward(self, pooled_output):
        seq_relationship_score = self.seq_relationship(pooled_output)
        return seq_relationship_score