(self, stage=None)
| 390 | instantiate_from_config(data_cfg) |
| 391 | |
| 392 | def setup(self, stage=None): |
| 393 | self.datasets = dict( |
| 394 | (k, instantiate_from_config(self.dataset_configs[k])) |
| 395 | for k in self.dataset_configs) |
| 396 | if self.wrap: |
| 397 | for k in self.datasets: |
| 398 | self.datasets[k] = WrappedDataset(self.datasets[k]) |
| 399 | |
| 400 | def _train_dataloader(self): |
| 401 | is_iterable_dataset = isinstance(self.datasets['train'], Txt2ImgIterableBaseDataset) |
no test coverage detected