(self, stage=None)
| 237 | return tform |
| 238 | |
| 239 | def setup(self, stage=None): |
| 240 | self.data_train = STL10( |
| 241 | root = self.data_dir, |
| 242 | split = 'train', |
| 243 | transform = self.transforms() |
| 244 | ) |
| 245 | |
| 246 | data_val_test = STL10( |
| 247 | root = self.data_dir, |
| 248 | split = 'test', |
| 249 | transform = self.transforms() |
| 250 | ) |
| 251 | test_size = int(0.7 * len(data_val_test)) |
| 252 | val_size = len(data_val_test) - test_size |
| 253 | self.data_val, self.data_test = random_split(data_val_test, [val_size, test_size]) |
| 254 | |
| 255 | def test_dataloader(self): |
| 256 | return DataLoader( |
nothing calls this directly
no test coverage detected