deepspeedai / deepspeedai/DeepSpeed
Deepspeed support CosineAnnealingLR scheduler
Open
Nobody has claimed this yet.
enhancement
- Dominant language
- Python
- Stars
- 43.1k
- Forks
- 5k
- Avg merge
- 4d 15h
- Merged PRs (30d)
- 112
Description
Now Deepspeed do not support CosineAnnealingLR scheduler.
So i want to support it by meself. question is how to develop a custom scheduler, Are there any tutorials available?
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by locating DeepSpeed's learning-rate scheduler integration and comparing it with PyTorch's CosineAnnealingLR documentation. Determine how custom schedulers are currently developed and tested, then define support for CosineAnnealingLR and verify that distributed training applies the expected schedule.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100