LinearSchedulerWithLinearWarmup

class lightning_ir.lightning_utils.schedulers.LinearSchedulerWithLinearWarmup(num_warmup_steps: int, num_training_steps: int, final_value: float = 0.0, num_delay_steps: int = 0, *args, **kwargs)

Bases: LambdaWarmupScheduler

__init__(num_warmup_steps: int, num_training_steps: int, final_value: float = 0.0, num_delay_steps: int = 0, *args, **kwargs) → None

Methods

__init__(num_warmup_steps, num_training_steps)
check_delay(current_step)
check_warmup(current_step)
value_lambda(current_step)