(self, shuffle=False)
| 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: |
| 459 | init_fn = worker_init_fn |
| 460 | else: |
| 461 | init_fn = None |
| 462 | return DataLoader(self.datasets["predict"], batch_size=self.batch_size, |
| 463 | num_workers=self.num_workers, worker_init_fn=init_fn) |
| 464 | |
| 465 | |
| 466 | class SetupCallback(Callback): |
nothing calls this directly
no outgoing calls
no test coverage detected