ray.rllib.utils.checkpoints.Checkpointable.set_state

abstract Checkpointable.set_state(state: Dict[str, Any]) → None

Sets the implementing class’ state to the given state dict.

If component keys are missing in state, these components of the implementing class will not be updated/set.

Parameters:

state – The state dict to restore the state from. Maps component keys to the corresponding subcomponent’s own state.