(self, batch, batch_idx)
| 95 | return F.log_softmax(x, dim=1) |
| 96 | |
| 97 | def training_step(self, batch, batch_idx): |
| 98 | x, y = batch |
| 99 | logits = self(x) |
| 100 | loss = F.nll_loss(logits, y) |
| 101 | return loss |
| 102 | |
| 103 | def validation_step(self, batch, batch_idx): |
| 104 | x, y = batch |
nothing calls this directly
no outgoing calls
no test coverage detected