[Feature] support pytorch CosineAnnealingWarmRestarts schedule

[bolin-chen]

What is the feature?

Want to use CosineAnnealingWarmRestarts scheduler. However, I cannot find this scheduler in mmengine. I also cannot find a tutorial about how to implement a custom scheduler myself. Could you give some tips about how to wrap a pytorch scheduler into mmengine？

Any other context?

No response