File size: 312 Bytes
59630ba | 1 2 3 4 5 6 7 8 9 10 | from lightning.pytorch.utilities.types import EVAL_DATALOADERS
from .base_data_module import BaseDataModule
class ValDataModule(BaseDataModule):
def val_dataloader(self) -> EVAL_DATALOADERS:
return [
self._dataloader(split) for split in self.exp_cfg.validation.dataset_splits
]
|