Method
forward
(self,
query_input_ids,
query_attention_mask,
context_input_ids,
context_attention_mask,
labels,
gck_segment=32)
Source from the content-addressed store, hash-verified
| 73 | return torch.cat(pooled_output, dim=0) |
| 74 | |
| 75 | def forward(self, |
| 76 | query_input_ids, |
| 77 | query_attention_mask, |
| 78 | context_input_ids, |
| 79 | context_attention_mask, |
| 80 | labels, |
| 81 | gck_segment=32): |
| 82 | query_vector = self.encode(self.qry_encoder, query_input_ids, |
| 83 | query_attention_mask, gck_segment) |
| 84 | context_vector = self.encode(self.ctx_encoder, context_input_ids, |
| 85 | context_attention_mask, gck_segment) |
| 86 | logits = torch.matmul(query_vector, context_vector.T) |
| 87 | loss = self.loss_fct(logits, labels) |
| 88 | return loss, logits |
| 89 | |
| 90 | |
| 91 | class ClassifyRerank(nn.Module): |
Callers
nothing calls this directly
Tested by
no test coverage detected