src.FRAME_FM.dataloaders.combined_dataloader

Classes

CombinedDataModule

A DataModule for combining datasets from multiple DataModules.

Module Contents

class src.FRAME_FM.dataloaders.combined_dataloader.CombinedDataModule(datamodules: list[src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule], batch_size: int = 32, num_workers: int = 4, pin_memory: bool = True, persistent_workers: bool = False, split_strategy: str = 'fraction', train_split: float = 0.85, val_split: float = 0.15, test_split: float = 0.0, train_transforms: Callable | None = None, val_transforms: Callable | None = None, test_transforms: Callable | None = None)[source]

Bases: src.FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModule

A DataModule for combining datasets from multiple DataModules. # TODO: Implement post-combination transforms

datamodules[source]