(self, loader_cfg: DataLoaderStageCfg)
| 54 | return None if loader_cfg.num_workers == 0 else loader_cfg.persistent_workers |
| 55 | |
| 56 | def get_generator(self, loader_cfg: DataLoaderStageCfg) -> torch.Generator | None: |
| 57 | if loader_cfg.seed is None: |
| 58 | return None |
| 59 | generator = Generator() |
| 60 | generator.manual_seed(loader_cfg.seed + self.global_rank) |
| 61 | return generator |
| 62 | |
| 63 | def train_dataloader(self): |
| 64 | dataset = get_dataset(self.dataset_cfgs, "train", self.frame_sampler_cfg) |
no outgoing calls
no test coverage detected