(self, stage=None)
| 189 | instantiate_from_config_sr(data_cfg) |
| 190 | |
| 191 | def setup(self, stage=None): |
| 192 | self.datasets = dict( |
| 193 | (k, instantiate_from_config_sr(self.dataset_configs[k])) |
| 194 | for k in self.dataset_configs) |
| 195 | if self.wrap: |
| 196 | for k in self.datasets: |
| 197 | self.datasets[k] = WrappedDataset(self.datasets[k]) |
| 198 | |
| 199 | def _train_dataloader(self): |
| 200 | is_iterable_dataset = isinstance(self.datasets['train'], Txt2ImgIterableBaseDataset) |
no test coverage detected