(
self,
dataset_cfgs: list[DatasetCfg],
data_module_cfg: DataModulePretrainCfg,
frame_sampler_cfg: FrameSamplerCfg,
global_rank: int,
)
| 38 | |
| 39 | class DataModulePretrain(LightningDataModule): |
| 40 | def __init__( |
| 41 | self, |
| 42 | dataset_cfgs: list[DatasetCfg], |
| 43 | data_module_cfg: DataModulePretrainCfg, |
| 44 | frame_sampler_cfg: FrameSamplerCfg, |
| 45 | global_rank: int, |
| 46 | ) -> None: |
| 47 | super().__init__() |
| 48 | self.dataset_cfgs = dataset_cfgs |
| 49 | self.data_module_cfg = data_module_cfg |
| 50 | self.frame_sampler_cfg = frame_sampler_cfg |
| 51 | self.global_rank = global_rank |
| 52 | |
| 53 | def get_persistent(self, loader_cfg: DataLoaderStageCfg) -> bool | None: |
| 54 | return None if loader_cfg.num_workers == 0 else loader_cfg.persistent_workers |
nothing calls this directly
no outgoing calls
no test coverage detected