Fix `CosineAnnealingWarmRestart` annotation (#61106)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/44770.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61106
Reviewed By: 1ntEgr8
Differential Revision: D29635764
Pulled By: walterddr
fbshipit-source-id: ddc45a7f04532a76d033ae7774706da1fa8608f7
