(self, stage)
| 50 | self.dataset_configs['test'] = test |
| 51 | |
| 52 | def setup(self, stage): |
| 53 | |
| 54 | if stage in ['fit']: |
| 55 | self.datasets = dict((k, instantiate_from_config(self.dataset_configs[k])) for k in self.dataset_configs) |
| 56 | else: |
| 57 | raise NotImplementedError |
| 58 | |
| 59 | def train_dataloader(self): |
| 60 |
no test coverage detected