Implement the cosine annealing learning rate schedule.
Signature: 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.
Math
Asked at
Output
Anything you print() in your code will show up here after you click Run.
Test Results