(self, input_ids, attention_mask)
| 28 | module.weight.data.normal_(mean=0.0, std=0.02) |
| 29 | |
| 30 | def query_emb(self, input_ids, attention_mask): |
| 31 | outputs1 = self.roberta(input_ids=input_ids, |
| 32 | attention_mask=attention_mask) |
| 33 | outputs1 = outputs1.last_hidden_state |
| 34 | full_emb = self.masked_mean_or_first(outputs1, attention_mask) |
| 35 | query1 = self.norm(self.embeddingHead(full_emb)) |
| 36 | return query1 |
| 37 | |
| 38 | |
| 39 | def doc_emb(self, input_ids, attention_mask): |
no test coverage detected