(self, stage=None)
| 105 | return tform |
| 106 | |
| 107 | def setup(self, stage=None): |
| 108 | self.data_train = self.dataclass( |
| 109 | root = f'{self.data_dir}/train', |
| 110 | transform = self.transforms(val=not self.augment) |
| 111 | ) |
| 112 | if self.num_samples is not None: |
| 113 | self.data_train = DataRepeater(self.data_train, self.num_samples) |
| 114 | |
| 115 | self.data_val = self.dataclass( |
| 116 | root = f'{self.data_dir}/val', |
| 117 | transform = self.transforms(val=True) |
| 118 | ) |
| 119 | |
| 120 | |
| 121 | class CubDataModule(_FGVCDataModule): |
nothing calls this directly
no test coverage detected