(self)
| 143 | dist.barrier() |
| 144 | |
| 145 | def create_dataloader(self): |
| 146 | config = self.config |
| 147 | ginfo = self.ginfo |
| 148 | |
| 149 | self.sampler = DistributedGivenIterationSampler( |
| 150 | self.dataset, config.max_iter, config.sampler.batch_size, |
| 151 | world_size=ginfo.task_size, rank=ginfo.task_rank, |
| 152 | last_iter=self.last_iter, shuffle_strategy=config.sampler.shuffle_strategy, |
| 153 | random_seed=ginfo.task_random_seed, ret_save_path=config.sampler.get('ret_save_path', None)) |
| 154 | self.loader = DataLoader(self.dataset, batch_size=config.sampler.batch_size, |
| 155 | shuffle=False, num_workers=config.workers, |
| 156 | pin_memory=False, sampler=self.sampler) |
| 157 | |
| 158 | def create_model(self): |
| 159 | config = self.config |
no test coverage detected