(self)
| 35 | return self.model(input_ids, **kwargs) |
| 36 | |
| 37 | def configure_optimizers(self): |
| 38 | optimizer = AdamW(self.parameters(), lr = self.learning_rate) |
| 39 | return optimizer |
| 40 | |
| 41 | def training_step(self, batch, batch_idx): |
| 42 | x, x_mask, y = batch |
nothing calls this directly
no outgoing calls
no test coverage detected