(self, batch, batch_idx)
| 41 | def training_step(self, batch, batch_idx): |
| 42 | x, y = batch |
| 43 | y_hat = self(x) |
| 44 | loss = F.cross_entropy(y_hat, y) |
| 45 | self.log('train_loss', loss) |
| 46 | return loss |
| 47 | |
| 48 | def validation_step(self, batch, batch_idx): |
| 49 | x, y = batch |
| 50 | y_hat = self(x) |
| 51 | loss = F.cross_entropy(y_hat, y) |
nothing calls this directly
no outgoing calls
no test coverage detected