(self, stage=None)
| 205 | instantiate_from_config(data_cfg) |
| 206 | |
| 207 | def setup(self, stage=None): |
| 208 | self.datasets = dict( |
| 209 | (k, instantiate_from_config(self.dataset_configs[k])) |
| 210 | for k in self.dataset_configs) |
| 211 | if self.wrap: |
| 212 | for k in self.datasets: |
| 213 | self.datasets[k] = WrappedDataset(self.datasets[k]) |
| 214 | |
| 215 | def _train_dataloader(self): |
| 216 | init_fn = None |
no test coverage detected