pytorch_pfn_extras.engine.StateObjectProtocol#

class pytorch_pfn_extras.engine.StateObjectProtocol(*args, **kwargs)#

Bases: Protocol

Methods

`__init__`(args, *kwargs)
`load_state_dict`(state_dict)
`state_dict`()

__init__(*args, **kwargs)#

load_state_dict(state_dict)#

Parameters:: state_dict (Dict[str, Any]) –
Return type:: None

state_dict()#

Return type:: Dict[str, Any]