ray.rllib.core.learner.learner_group.LearnerGroup.get_weights#
- LearnerGroup.get_weights(module_ids: Set[str] | None = None, inference_only: bool = False) Dict[str, Any] [source]#
Get the weights of the MultiAgentRLModule maintained by each Learner.
- Parameters:
module_ids – The ids of the modules to get the weights of.
- Returns:
A mapping of module ids to their weights.