(self)
| 59 | self.worker_init_fn = None |
| 60 | |
| 61 | def create_dataloader(self): |
| 62 | config = self.config |
| 63 | ginfo = self.ginfo |
| 64 | |
| 65 | self.sampler = DistributedGivenIterationSampler( |
| 66 | self.dataset, config.max_iter, config.sampler.batch_size, |
| 67 | world_size=ginfo.task_size, rank=ginfo.task_rank, |
| 68 | last_iter=self.last_iter, |
| 69 | shuffle_strategy=config.sampler.shuffle_strategy, |
| 70 | random_seed=ginfo.task_random_seed, |
| 71 | ret_save_path=config.sampler.get('ret_save_path', None)) |
| 72 | |
| 73 | |
| 74 | self.loader = DataLoader(self.dataset, batch_size=config.sampler.batch_size, |
| 75 | shuffle=False, num_workers=config.workers, |
| 76 | pin_memory=False, sampler=self.sampler, worker_init_fn=self.worker_init_fn) |
nothing calls this directly
no test coverage detected