ray.rllib.core.learner.learner.Learner.get_优化器_for_模块#

Learner.get_optimizers_for_module(module_id: str = '__all_modules__') List[Tuple[str, torch.optim.Optimizer | tf.keras.optimizers.Optimizer]][源代码]#

返回一个 (优化器名称, 优化器实例) 元组列表,对应于 module_id。

参数:

module_id – 要返回其配置的(优化器名称,优化器)对的 ModuleID。如果未提供,将返回在 ALL_MODULES 下注册的优化器。

返回:

([优化器名称], [优化器对象]),其中优化器名称是优化器在 self.register_optimizer 中注册的名称。如果只为 module_id 配置了一个优化器,[优化器名称] 将是 DEFAULT_OPTIMIZER。

返回类型:

A list of tuples of the format