src.FRAME_FM.dataloaders.demo_eurosat
Classes
FRAME-FM DataModule for EuroSAT. |
Module Contents
- class src.FRAME_FM.dataloaders.demo_eurosat.EuroSATDataModule(data_root: str = 'data', batch_size: int = 32, num_workers: int = 4, pin_memory: bool = True, persistent_workers: bool = False, **kwargs: Any)[source]
Bases:
FRAME_FM.utils.LightningDataModuleWrapper.BaseDataModuleFRAME-FM DataModule for EuroSAT.
Uses BaseDataModule for split logic (split_strategy, indices/fractions).
Uses Hydra-provided transforms (train_transforms, val_transforms, test_transforms).
Wraps split datasets in TransformedDataset so each split can have its own transform.