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

RLModule.save_state(dir: str | Path) None[source]#

Saves the weights of this RLModule to the directory dir.

Parameters:

dir – The directory to save the checkpoint to.