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