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

LearnerGroup.set_weights(weights: Mapping[str, Any]) → None[source]#

Set the weights of the MultiAgentRLModule maintained by each Learner.

The weights don’t have to include all the modules in the MARLModule.

This way the weights of only some of the Agents can be set.

Parameters

weights – The weights to set each RLModule in the MARLModule to.