TrainingBatchLoop¶

class pytorch_lightning.loops.batch.TrainingBatchLoop[source]¶

Bases: pytorch_lightning.loops.loop.Loop[List[Union[Dict[int, Dict[str, Any]], Dict[str, Any]]]]

Runs over a single batch of data.

advance(kwargs)[source]¶

Runs the train step together with optimization (if necessary) on the current batch split.

Parameters: kwargs¶ (OrderedDict) – the kwargs passed down to the hooks.
Return type: None

connect(optimizer_loop=None, manual_loop=None)[source]¶

Optionally connect one or multiple loops to this one.

Linked loops should form a tree.

on_run_end()[source]¶

Hook to be called at the end of the run.

Its return argument is returned from run.

Return type: List[Union[Dict[int, Dict[str, Any]], Dict[str, Any]]]

on_run_start(kwargs)[source]¶

Splits the data into tbptt splits.

Parameters: kwargs¶ (OrderedDict) – the kwargs passed down to the hooks.
Return type: None

Resets the loop state.

Use to release memory etc.

property done: bool¶

Returns if all batch splits have been processed already.