set_weights

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

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

Parameters:

weights (Dict[str, Any]) – The weights dict of the MultiRLModule of a Learner inside this LearnerGroup.