(self, config)
| 1115 | ) |
| 1116 | class BertForNextSentencePrediction(BertPreTrainedModel): |
| 1117 | def __init__(self, config): |
| 1118 | super().__init__(config) |
| 1119 | |
| 1120 | self.bert = BertModel(config) |
| 1121 | self.cls = BertOnlyNSPHead(config) |
| 1122 | |
| 1123 | self.init_weights() |
| 1124 | |
| 1125 | @add_start_docstrings_to_callable(BERT_INPUTS_DOCSTRING.format("(batch_size, sequence_length)")) |
| 1126 | def forward( |
nothing calls this directly
no test coverage detected