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

LearnerGroup.set_weights(weights: Dict[str, Any]) → None

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.