(self, stage=None)
| 73 | pass |
| 74 | |
| 75 | def setup(self, stage=None): |
| 76 | self.datasets = dict((k, instantiate_from_config(self.dataset_configs[k])) for k in self.dataset_configs) |
| 77 | if self.wrap: |
| 78 | for k in self.datasets: |
| 79 | self.datasets[k] = WrappedDataset(self.datasets[k]) |
| 80 | |
| 81 | def _train_dataloader(self): |
| 82 | is_iterable_dataset = isinstance(self.datasets['train'], Txt2ImgIterableBaseDataset) |
no test coverage detected