(self)
| 28 | return self.model(input_ids, **kwargs) |
| 29 | |
| 30 | def configure_optimizers(self): |
| 31 | optimizer = AdamW(self.parameters(), lr = self.learning_rate) |
| 32 | return optimizer |
| 33 | |
| 34 | def training_step(self, batch, batch_idx): |
| 35 | pos_ids, pos_mask, neg_ids, neg_mask = batch |
nothing calls this directly
no outgoing calls
no test coverage detected