espnet2.schedulers.piecewise_linear_warmup_lr.PiecewiseLinearWarmupLR

source

class espnet2.schedulers.piecewise_linear_warmup_lr.PiecewiseLinearWarmupLR(optimizer: Optimizer, warmup_steps_list: List[float | int] = [0, 25000], warmup_lr_list: List[float] = [0.0, 0.001], last_epoch: int = -1)

Bases: _LRScheduler, AbsBatchStepScheduler

The PiecewiseLinearWarmupLR scheduler

This scheduler is similar to WarmupLR Scheduler except that the warmup stage is piecewise linear.

get_lr()