(self, shuffle=False)
| 431 | worker_init_fn=init_fn) |
| 432 | |
| 433 | def _val_dataloader(self, shuffle=False): |
| 434 | if isinstance(self.datasets['validation'], Txt2ImgIterableBaseDataset) or self.use_worker_init_fn: |
| 435 | init_fn = worker_init_fn |
| 436 | else: |
| 437 | init_fn = None |
| 438 | return DataLoader(self.datasets["validation"], |
| 439 | batch_size=self.batch_size, |
| 440 | num_workers=self.num_workers, |
| 441 | worker_init_fn=init_fn, |
| 442 | shuffle=shuffle) |
| 443 | |
| 444 | def _test_dataloader(self, shuffle=False): |
| 445 | is_iterable_dataset = isinstance(self.datasets['train'], Txt2ImgIterableBaseDataset) |
nothing calls this directly
no outgoing calls
no test coverage detected