| 365 | |
| 366 | |
| 367 | class ConcatDataset(Dataset): |
| 368 | def __init__(self, *datasets): |
| 369 | self.datasets = datasets |
| 370 | |
| 371 | def __getitem__(self, idx): |
| 372 | return tuple(d[idx] for d in self.datasets) |
| 373 | |
| 374 | def __len__(self): |
| 375 | return min(len(d) for d in self.datasets) |
| 376 | |
| 377 | |
| 378 | class DataModuleFromConfig(pl.LightningDataModule): |