Merge pull request #1 from FluxML/master

fix AdamW (#198)
FluxML · Dec 11, 2024 · d358026 · d358026
2 parents acbe8e3 + 669798c
commit d358026
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rules.jl b/src/rules.jl
@@ -531,8 +531,8 @@ Implemented as an [`OptimiserChain`](@ref) of [`Adam`](@ref) and [`WeightDecay`]
 struct AdamW{T1,T2,T3,T4} <: AbstractRule
   eta::T1
   beta::T2
-  epsilon::T3
   lambda::T4
+  epsilon::T3
   couple::Bool
 end