(self, stage=None)
| 382 | instantiate_from_config(data_cfg) |
| 383 | |
| 384 | def setup(self, stage=None): |
| 385 | self.datasets = dict( |
| 386 | (k, instantiate_from_config(self.dataset_configs[k])) |
| 387 | for k in self.dataset_configs) |
| 388 | if self.wrap: |
| 389 | for k in self.datasets: |
| 390 | self.datasets[k] = WrappedDataset(self.datasets[k]) |
| 391 | |
| 392 | def _train_dataloader(self): |
| 393 | is_iterable_dataset = isinstance(self.datasets['train'], Txt2ImgIterableBaseDataset) |