Serializable#

class composer.core.Serializable[source]#

Interface for serialization; used by checkpointing.

load_state_dict(state)[source]#

Restores the state of the object.

Parameters

state (dict[str, Any]) โ€“ The state of the object, as previously returned by state_dict().

state_dict()[source]#

Returns a dictionary representing the internal state.

The returned dictionary must be pickale-able via torch.save().

Returns

dict[str, Any] โ€“ The state of the object.