ray.rllib.core.learner.learner.Learner.get_优化器

Learner.get_optimizer(module_id: str = 'default_policy', optimizer_name: str = 'default_optimizer') → torch.optim.Optimizer | tf.keras.optimizers.Optimizer

返回在给定的 module_id 和名称下配置的优化器对象。

如果在 module_id`（或 ALL_MODULES）下仅通过 `self.register_optimizer 方法注册了一个优化器，则假定 optimizer_name 为 DEFAULT_OPTIMIZER。

参数:

• module_id – 要返回其配置优化器的 ModuleID。如果未提供，将假设为 DEFAULT_MODULE_ID。

• optimizer_name – 要返回的优化器名称（通过 self.register_optimizer() 在 module_id 下注册）。如果未提供，将假定为 DEFAULT_OPTIMIZER。

返回:

优化器对象，在给定的 module_id 和 optimizer_name 下配置。