lightning_template.utils.optim.keep_and_linearly_decay_lr_scheduler

Classes

KeepAndLinearlyDecayLrScheduler

Keep lr first then decay learning rate linearly in optimizer.

Module Contents

class lightning_template.utils.optim.keep_and_linearly_decay_lr_scheduler.KeepAndLinearlyDecayLrScheduler(optimizer, keep_epochs, decay_epochs, last_epoch=-1, verbose=False)

Bases: torch.optim.lr_scheduler._LRScheduler

Keep lr first then decay learning rate linearly in optimizer.

Parameters:

• optimizer (Optimizer) – Wrapped optimizer.

• keep_epochs – target keep epoch.

• decay_epochs – target decay epoch.

keep_epochs

decay_epochs

get_lr()

Compute learning rate using chainable form of the scheduler.