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

restrict AdamW's fields to Float64 #199

Merged
merged 2 commits into from
Dec 12, 2024
Merged

Conversation

CarloLucibello
Copy link
Member

As per comment #198 (comment),
since we do it for any other rule (explicitly or through @def) except for Descent.

Is there any reason why Descent is left out @mcabbott ?

src/rules.jl Outdated
epsilon::T3
struct AdamW <: AbstractRule
eta::Float64
beta::Float64
Copy link
Member

Choose a reason for hiding this comment

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

This won't work, function AdamW(η, β = (0.9, 0.999), λ = 0.0...

@mcabbott
Copy link
Member

reason why Descent is left out

Yes, someone complained that they had a use case for Descent with complex eta. Maybe discussed in the original PR.

src/rules.jl Outdated Show resolved Hide resolved
@CarloLucibello CarloLucibello merged commit 5f2680a into master Dec 12, 2024
4 of 5 checks passed
@mcabbott mcabbott deleted the CarloLucibello-patch-1 branch December 12, 2024 18: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