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

abstract static Learner._set_optimizer_lr(optimizer: torch.optim.Optimizer | tf.keras.optimizers.Optimizer, lr: float) → None[source]#

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.