(self, stage=None)
| 168 | instantiate_from_config(data_cfg) |
| 169 | |
| 170 | def setup(self, stage=None): |
| 171 | self.datasets = dict( |
| 172 | (k, instantiate_from_config(self.dataset_configs[k])) |
| 173 | for k in self.dataset_configs) |
| 174 | if self.wrap: |
| 175 | for k in self.datasets: |
| 176 | self.datasets[k] = WrappedDataset(self.datasets[k]) |
| 177 | |
| 178 | def _train_dataloader(self): |
| 179 | return DataLoader(self.datasets["train"], batch_size=self.batch_size, |