PredictionEpochLoop

class pytorch_lightning.loops.epoch.PredictionEpochLoop

Bases: pytorch_lightning.loops.loop.Loop

Loop performing prediction on arbitrary sequentially used dataloaders.

advance(dataloader_iter, dataloader_idx, dl_max_batches, num_dataloaders)

Runs one prediction step.

Parameters:

• dataloader_iter (Iterator) – the iterator over the current dataloader

• dataloader_idx (int) – the index of the current dataloader

• dl_max_batches (int) – the maximum number of batches the current loader can produce

• num_dataloaders (int) – the total number of dataloaders

Return type:

None

connect(**kwargs)

Optionally connect one or multiple loops to this one.

Linked loops should form a tree.

Return type:

None

on_run_end()

Returns the predictions and the corresponding batch indices.

Return type:

Tuple[List[Any], List[List[int]]]

on_run_start(dataloader_iter, dataloader_idx, dl_max_batches, num_dataloaders)

Prepares the loops internal state.

Parameters:

• dataloader_iter (Iterator) – the iterator over the current dataloader

• dataloader_idx (int) – the index of the current dataloader

• dl_max_batches (int) – the maximum number of batches the current loader can produce

• num_dataloaders (int) – the total number of dataloaders

Return type:

None

reset()

Resets the loops internal state.

Return type:

None

property done: bool

Ends prediction when the iteration count exceeds the total number of available batches.

property should_store_predictions: bool

Whether the predictions should be stored for later usage (e.g. aggregation or returning)