mllooper.LooperIterationStop#

class mllooper.LooperIterationStop(step_iteration_limit: Optional[int] = None, total_iteration_limit: Optional[int] = None, **kwargs)[source]#

Bases: Module

load_state_dict(state_dict: Dict[str, any], strict: bool = True)[source]#

Load the modules state from a dictionary.

Parameters:

  • state_dict (Dict[str, any]) – The state dictionary to load

  • strict (bool) – If true rise an error on missing or additional keys in the state dict. If false these keys will be ignored.

state_dict() → Dict[str, Any][source]#

Return the state of the module as dictionary.

All items of the dictionary should be serializable by pickle.

Returns:

The modules current state as dictionary

Return type:

Dict[str, Any]

step(state: State) → None[source]#

Perform a step of the module on the state.

Parameters:

state (State) – The current state