(self, stage=None)
| 165 | return tform |
| 166 | |
| 167 | def setup(self, stage=None): |
| 168 | self.data_train = self.dataclass( |
| 169 | root = self.data_dir, |
| 170 | train = True, |
| 171 | transform = self.transforms(not self.augment) |
| 172 | ) |
| 173 | self.data_val = self.dataclass( |
| 174 | root = self.data_dir, |
| 175 | train = False, |
| 176 | transform = self.transforms(True) |
| 177 | ) |
| 178 | |
| 179 | |
| 180 | class Cifar10DataModule(_CifarDataModule): |
nothing calls this directly
no test coverage detected