ConstantSchedulerWithQuadraticWarmup

class lightning_ir.lightning_utils.schedulers.ConstantSchedulerWithQuadraticWarmup(num_warmup_steps: int, num_delay_steps: int = 0, *args, **kwargs)

Bases: LambdaWarmupScheduler

__init__(num_warmup_steps: int, num_delay_steps: int = 0, *args, **kwargs) → None

Methods

__init__(num_warmup_steps[, num_delay_steps])
check_delay(current_step)
check_warmup(current_step)
value_lambda(current_step)