-
Notifications
You must be signed in to change notification settings - Fork 28.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add RAdamScheduleFree optimizer #35313
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for this contribution @nhamanasu ! Left a few comments
Thank you for the valuable feedback! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice, thanks for iterating !
Hi guys, is anyone working on this? |
I think I did everything I could do for merging this PR, and just waiting for the approval from @ArthurZucker (sorry for mentioning...) |
Could you just fix the conflits ? I will merge this right after ! |
Thanks for the fix, merging as the failing test is unrelated! |
* add RAdamScheduleFree optimizer * revert schedulefree version to the minimum requirement * refine is_schedulefree_available so that it can take min_version * refine documents --------- Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
* add RAdamScheduleFree optimizer * revert schedulefree version to the minimum requirement * refine is_schedulefree_available so that it can take min_version * refine documents --------- Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
Sorry for the delayed follow-up, and thank you for taking care of the final fix to get this PR merged. |
What does this PR do?
Integrates the new RAdamScheduleFree optimizer in meta's schedule_free.
This PR has greatly followed the previous PR to introduce schedule_free #30079
More details
test_trainer.py
& confirmed the test passesBackgrounds
I recently made the RAdamScheduleFree optimizer, and it's been merged in schedule_free library since version 1.4.
RAdamScheduleFree eliminates:
Before submitting
Pull Request section?
to it if that's the case.
documentation guidelines, and
here are tips on formatting docstrings.
Who can review?
Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.
@SunMarc @ydshieh @muellerzr