ray.rllib.core.rl_module.rl_module.RLModule.save_state#

RLModule.save_state(dir: Union[str, pathlib.Path]) → None[source]#

Saves the weights of this RLModule to the directory dir.

Parameters

dir – The directory to save the checkpoint to.