LambdaWarmupScheduler

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

Bases: ABC

__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)