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

Set the state of the learner.


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.