Skip to content
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

Implement AdaBelief #1362

Merged
merged 6 commits into from
Oct 19, 2020
Merged

Implement AdaBelief #1362

merged 6 commits into from
Oct 19, 2020

Conversation

willtebbutt
Copy link
Member

@willtebbutt willtebbutt commented Oct 16, 2020

Implements the fancy new AdaBelief algorithm. I guess time will tell whether or not it's helpful, but it doesn't do any harm including it, and it's a pretty trivial change from ADAM.

PR Checklist

  • Tests are added
  • Entry in NEWS.md
  • Documentation, if applicable
  • Final review from @dhairyagandhi96 (for API changes).

edit: I've tweaked the docstring from ADAM, which I think probably counts as documentation as I'm assuming it's going to get autodoc-ed into the optimiser documentation.

@DhairyaLGandhi
Copy link
Member

Needs to be exported from the Opitmise module. Otherwise lgtm

@willtebbutt
Copy link
Member Author

@DhairyaLGandhi looks like things are passing. Are you happy to approve?

Copy link
Member

@DhairyaLGandhi DhairyaLGandhi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, thanks so much!

bors r+

@bors
Copy link
Contributor

bors bot commented Oct 19, 2020

Build succeeded:

@bors bors bot merged commit c5c35cc into FluxML:master Oct 19, 2020
@willtebbutt willtebbutt deleted the wct/adabelief branch October 22, 2020 22:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants