(self, shuffle=False)
| 442 | shuffle=shuffle) |
| 443 | |
| 444 | def _test_dataloader(self, shuffle=False): |
| 445 | is_iterable_dataset = isinstance(self.datasets['train'], Txt2ImgIterableBaseDataset) |
| 446 | if is_iterable_dataset or self.use_worker_init_fn: |
| 447 | init_fn = worker_init_fn |
| 448 | else: |
| 449 | init_fn = None |
| 450 | |
| 451 | # do not shuffle dataloader for iterable dataset |
| 452 | shuffle = shuffle and (not is_iterable_dataset) |
| 453 | |
| 454 | return DataLoader(self.datasets["test"], batch_size=self.batch_size, |
| 455 | num_workers=self.num_workers, worker_init_fn=init_fn, shuffle=shuffle) |
| 456 | |
| 457 | def _predict_dataloader(self, shuffle=False): |
| 458 | if isinstance(self.datasets['predict'], Txt2ImgIterableBaseDataset) or self.use_worker_init_fn: |
nothing calls this directly
no outgoing calls
no test coverage detected