ray.rllib.core.learner.learner_group.LearnerGroup.get_weights

LearnerGroup.get_weights(module_ids: Collection[str] | None = None) → Dict[str, Any]

Convenience method instead of self.get_state(components=…).

Parameters:

module_ids – An optional collection of ModuleIDs for which to return weights. If None (default), return weights of all RLModules.

Returns:

The results of self.get_state(components='learner/rl_module')['learner']['rl_module'].