.. _LambdaLR:

LambdaLR
========

class ``modelzoo.common.pytorch.optim.lr_scheduler.LambdaLR`` (optimizer: torch.optim.optimizer.Optimizer, initial_learning_rate: float, disable_lr_steps_reset: bool = False)

Sets the learning rate of each parameter group to the initial lr times a given function (which is specified by overriding ``set_lr_lambda``).

Parameters:
  **optimizer** – The optimizer to schedule

  **initial_learning_rate** – The initial learning rate