Warmup + Cosine Decay Schedule

Unlock premium problems

This is a Premium problem. Upgrade to unlock the full editor, test cases, and solution.