ray.rllib.core.learner.learner.Learner._set_optimizer_lr

abstract static Learner._set_optimizer_lr(optimizer: Union[torch.optim.optimizer.Optimizer, keras.optimizers.optimizer_experimental.optimizer.Optimizer], lr: float) → None

Updates the learning rate of the given local optimizer.

Parameters

• optimizer – The local optimizer to update the learning rate for.

• lr – The new learning rate.