ray.rllib.core.rl_module.rl_module.RLModule.load_state

RLModule.load_state(dir: Union[str, pathlib.Path]) → None

Loads the weights of an RLModule from the directory dir.

Parameters

dir – The directory to load the checkpoint from.