Perform a forward pass through the model `self.net`. :param x: A tensor of images. :return: A tensor of logits.
(self, x: torch.Tensor)
| 77 | self.val_acc_best = MaxMetric() |
| 78 | |
| 79 | def forward(self, x: torch.Tensor) -> torch.Tensor: |
| 80 | """Perform a forward pass through the model `self.net`. |
| 81 | |
| 82 | :param x: A tensor of images. |
| 83 | :return: A tensor of logits. |
| 84 | """ |
| 85 | return self.net(x) |
| 86 | |
| 87 | def on_train_start(self) -> None: |
| 88 | """Lightning hook that is called when training begins.""" |