(self)
| 110 | return loss |
| 111 | |
| 112 | def configure_optimizers(self): |
| 113 | optimizer = torch.optim.Adam(self.parameters(), lr=self.learning_rate) |
| 114 | return optimizer |
| 115 | |
| 116 | def prepare_data(self): |
| 117 | MNIST(self.data_dir, train=True, download=True) |
nothing calls this directly
no outgoing calls
no test coverage detected