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.