MCPcopy Create free account
hub / github.com/apple/axlearn / set_state

Method set_state

axlearn/common/checkpointer.py:269–273  ·  view source on GitHub ↗

Sets checkpoint state. The input is a `msgpack`-deserialized object corresponding to restored checkpoint state.

(self, state: Union[bytes, Nested[Any]])

Source from the content-addressed store, hash-verified

267 """
268
269 def set_state(self, state: Union[bytes, Nested[Any]]):
270 """Sets checkpoint state.
271
272 The input is a `msgpack`-deserialized object corresponding to restored checkpoint state.
273 """
274
275
276# pylint: disable-next=redefined-builtin

Callers 3

_test_checkpointingMethod · 0.45
_test_checkpointingMethod · 0.45

Calls

no outgoing calls

Tested by 2

_test_checkpointingMethod · 0.36
_test_checkpointingMethod · 0.36