(self, stage=None)
| 118 | MNIST(self.data_dir, train=False, download=True) |
| 119 | |
| 120 | def setup(self, stage=None): |
| 121 | self.mnist_train = MNIST(self.data_dir, train=True, transform=self.transform) |
| 122 | self.mnist_val = MNIST(self.data_dir, train=False, transform=self.transform) |
| 123 | |
| 124 | def train_dataloader(self): |
| 125 | return DataLoader(self.mnist_train, batch_size=self.batch_size) |
nothing calls this directly
no outgoing calls
no test coverage detected