mipcandy.common.optim.lr_scheduler#

Module Contents#

Classes#

API#

class mipcandy.common.optim.lr_scheduler.AbsoluteLinearLR(optimizer: torch.optim.Optimizer, k: float, b: float, *, min_lr: float = 1e-06, restart: bool = False, last_epoch: int = -1)[source]#

Bases: torch.optim.lr_scheduler.LRScheduler

lr = kx + b

Initialization

_interp(epoch: int) float[source]#
get_lr() list[float][source]#
class mipcandy.common.optim.lr_scheduler.PolyLRScheduler(optimizer: torch.optim.Optimizer, initial_lr: float, max_steps: int, *, exponent: float = 0.9, last_epoch: int = -1)[source]#

Bases: torch.optim.lr_scheduler.LRScheduler

_interp(epoch: int) float[source]#
get_lr() list[float][source]#