(self, stage=None)
| 405 | instantiate_from_config(data_cfg) |
| 406 | |
| 407 | def setup(self, stage=None): |
| 408 | self.datasets = dict( |
| 409 | (k, instantiate_from_config(self.dataset_configs[k])) |
| 410 | for k in self.dataset_configs) |
| 411 | if self.wrap: |
| 412 | for k in self.datasets: |
| 413 | self.datasets[k] = WrappedDataset(self.datasets[k]) |
| 414 | |
| 415 | def _train_dataloader(self): |
| 416 | is_iterable_dataset = isinstance(self.datasets['train'], Txt2ImgIterableBaseDataset) |