| 342 | |
| 343 | |
| 344 | class ConcatDataset(Dataset): |
| 345 | def __init__(self, *datasets): |
| 346 | self.datasets = datasets |
| 347 | |
| 348 | def __getitem__(self, idx): |
| 349 | return tuple(d[idx] for d in self.datasets) |
| 350 | |
| 351 | def __len__(self): |
| 352 | return min(len(d) for d in self.datasets) |
| 353 | |
| 354 | |
| 355 | class DataModuleFromConfig(pl.LightningDataModule): |