Fix type annotation for CosineAnnealingLR (#41866)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/41866
Reviewed By: izdeby
Differential Revision: D22703576
Pulled By: mrshenli
fbshipit-source-id: 10a0f593ffaaae82a2923a42815c36793a9043d5
