PredictionLoop
- class pytorch_lightning.loops.dataloader.PredictionLoop[source]
Bases:
pytorch_lightning.loops.dataloader.dataloader_loop.DataLoaderLoop
Loop to run over dataloaders for prediction.
- on_run_end()[source]
Calls
on_predict_epoch_end
andon_predict_end
hooks and returns results from all dataloaders.
- property dataloaders: Sequence[torch.utils.data.dataloader.DataLoader]
Returns all prediction dataloaders.
- Return type
- property max_batches: List[int]
The max number of batches this loop will run for each dataloader.