(self, sequence_output, pooled_output)
| 534 | self.seq_relationship = nn.Linear(config.hidden_size, 2) |
| 535 | |
| 536 | def forward(self, sequence_output, pooled_output): |
| 537 | prediction_scores = self.predictions(sequence_output) |
| 538 | seq_relationship_score = self.seq_relationship(pooled_output) |
| 539 | return prediction_scores, seq_relationship_score |
| 540 | |
| 541 | |
| 542 | class BertPreTrainedModel(PreTrainedModel): |
nothing calls this directly
no outgoing calls
no test coverage detected