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

Learner.set_state(state: Mapping[str, Any]) → None[source]#

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.