(self, batch, batch_idx)
| 101 | return loss |
| 102 | |
| 103 | def validation_step(self, batch, batch_idx): |
| 104 | x, y = batch |
| 105 | logits = self(x) |
| 106 | loss = F.nll_loss(logits, y) |
| 107 | preds = torch.argmax(logits, dim=1) |
| 108 | acc = multiclass_accuracy(preds, y, num_classes=self.num_classes) |
| 109 | self.log("val_acc", acc, prog_bar=False) |
| 110 | return loss |
| 111 | |
| 112 | def configure_optimizers(self): |
| 113 | optimizer = torch.optim.Adam(self.parameters(), lr=self.learning_rate) |
nothing calls this directly
no outgoing calls
no test coverage detected