ray.rllib.core.learner.learner.Learner.set_state

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

Set the state of the learner.

Parameters:

state – The state of the optimizer and module. Can be obtained from get_state. State is a dictionary with two keys: “module_state” and “optimizer_state”. The value of each key is a dictionary that can be passed to set_module_state and set_optimizer_state respectively.