(self, stage=None)
| 637 | self.cfg = parse_structured(MultiObjectDataModuleConfig, cfg) |
| 638 | |
| 639 | def setup(self, stage=None) -> None: |
| 640 | if stage in [None, "fit"]: |
| 641 | self.train_dataset = MultiObjectDataset(self.cfg, "train") |
| 642 | if stage in [None, "fit", "validate"]: |
| 643 | self.val_dataset = MultiObjectDataset(self.cfg, "val") |
| 644 | if stage in [None, "test", "predict"]: |
| 645 | self.test_dataset = MultiObjectDataset(self.cfg, "test") |
| 646 | |
| 647 | def prepare_data(self): |
| 648 | pass |
no test coverage detected