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