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

LearnerGroup.set_weights(weights) → None

Convenience method instead of self.set_state({‘learner’: {‘rl_module’: ..}}).

Parameters:

weights – The weights dict of the MARLModule of a Learner inside this LearnerGroup.