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

MultiAgentRLModule.load_state(path: str | Path, modules_to_load: Set[str] | None = 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.