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

LearnerGroup.set_weights(weights) None[source]#

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

Parameters:

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