(self, shuffle=False)
| 408 | worker_init_fn=init_fn) |
| 409 | |
| 410 | def _val_dataloader(self, shuffle=False): |
| 411 | if isinstance(self.datasets['validation'], Txt2ImgIterableBaseDataset) or self.use_worker_init_fn: |
| 412 | init_fn = worker_init_fn |
| 413 | else: |
| 414 | init_fn = None |
| 415 | return DataLoader(self.datasets["validation"], |
| 416 | batch_size=self.batch_size, |
| 417 | num_workers=self.num_workers, |
| 418 | worker_init_fn=init_fn, |
| 419 | shuffle=shuffle) |
| 420 | |
| 421 | def _test_dataloader(self, shuffle=False): |
| 422 | is_iterable_dataset = isinstance(self.datasets['train'], Txt2ImgIterableBaseDataset) |
nothing calls this directly
no outgoing calls
no test coverage detected