ray.rllib.core.learner.learner_group.LearnerGroup.get_weights#

LearnerGroup.get_weights(module_ids: Optional[Set[str]] = None) → Mapping[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.