ray.rllib.core.rl_module.marl_module.MultiAgentRLModule.load_state#

MultiAgentRLModule.load_state(path: Union[str, pathlib.Path], modules_to_load: Optional[Set[str]] = None) → None[source]#

Loads the weights of an MultiAgentRLModule from dir.

Note

If you want to load a module that is not already in this MultiAgentRLModule, you should add it to this MultiAgentRLModule before loading the checkpoint.

Parameters

  • path – The path to the directory to load the state from.

  • modules_to_load – The modules whose state is to be loaded from the path. If this is None, all modules that are checkpointed will be loaded into this marl module.