Implement the cosine annealing learning rate schedule.
Signature: def cosine_lr(t: int, T: int, lr_max: float, lr_min: float = 0.0) -> float
def cosine_lr(t: int, T: int, lr_max: float, lr_min: float = 0.0) -> float
Return the learning rate at step t given total steps T.
t
T
Math
Asked at
Test Results