ray.rllib.core.learner.learner_group.LearnerGroup.__init__

LearnerGroup.__init__(*, config: AlgorithmConfig, module_spec: RLModuleSpec | MultiRLModuleSpec | None = None)

初始化一个 LearnerGroup 实例。

参数:

• config – 用于配置此 LearnerGroup 的 AlgorithmConfig 对象。调用配置上的 learners(num_learners=...) 方法来指定要使用的学习者工作者的数量。使用参数 num_cpus_per_learner 和/或 num_gpus_per_learner 调用相同的方法来配置此 LearnerGroup 中每个 Learner 工作者使用的计算资源。调用配置上的 training(learner_class=...) 方法来指定要使用的具体 Learner 类。调用配置上的 rl_module(rl_module_spec=...) 方法来设置每个 Learner 中使用的 RLModule 的具体细节。

• module_spec – 如果尚未在 config 中指定，可以通过此参数提供一个单独的重写 RLModuleSpec。