ray.rllib.core.learner.learner_group.LearnerGroup.获取权重

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

便捷方法，替代 self.get_state(components=…)。

参数:

module_ids – 一个可选的 ModuleID 集合，用于返回权重。如果为 None（默认），则返回所有 RLModules 的权重。

返回:

self.get_state(components='learner/rl_module')['learner']['rl_module'] 的结果。