baselooper.LooperIterationStop

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

Bases: baselooper.module.Module

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

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]

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: baselooper.state.State) → None

Perform a step of the module on the state.

Parameters

state (State) – The current state