(self, shuffle=False)
| 432 | num_workers=self.num_workers, worker_init_fn=init_fn, shuffle=shuffle) |
| 433 | |
| 434 | def _predict_dataloader(self, shuffle=False): |
| 435 | if isinstance(self.datasets['predict'], Txt2ImgIterableBaseDataset) or self.use_worker_init_fn: |
| 436 | init_fn = worker_init_fn |
| 437 | else: |
| 438 | init_fn = None |
| 439 | return DataLoader(self.datasets["predict"], batch_size=self.batch_size, |
| 440 | num_workers=self.num_workers, worker_init_fn=init_fn) |
| 441 | |
| 442 | |
| 443 | class SetupCallback(Callback): |
nothing calls this directly
no outgoing calls
no test coverage detected