| 347 | |
| 348 | |
| 349 | class FakeDataloader: |
| 350 | def __init__(self, num_iters): |
| 351 | self.num_iters = num_iters |
| 352 | |
| 353 | def __iter__(self): |
| 354 | if self.num_iters is not None: |
| 355 | for _ in range(self.num_iters): |
| 356 | yield None |
| 357 | else: |
| 358 | while True: |
| 359 | yield None |
| 360 | |
| 361 | |
| 362 | def build_data_loader(dataset, batch_size, num_workers, drop_last, shuffle=True, only_rank0=False): |