(self, loader_cfg: DataLoaderStageCfg)
| 78 | self.global_rank = global_rank |
| 79 | |
| 80 | def get_persistent(self, loader_cfg: DataLoaderStageCfg) -> bool | None: |
| 81 | return None if loader_cfg.num_workers == 0 else loader_cfg.persistent_workers |
| 82 | |
| 83 | def get_generator(self, loader_cfg: DataLoaderStageCfg) -> torch.Generator | None: |
| 84 | if loader_cfg.seed is None: |
no outgoing calls