Skip to content

Commit 3c046a1

Browse files
change fields order
1 parent ba7cafc commit 3c046a1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/rules.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -531,14 +531,14 @@ Implemented as an [`OptimiserChain`](@ref) of [`Adam`](@ref) and [`WeightDecay`]
531531
struct AdamW{T1,T2,T3,T4} <: AbstractRule
532532
eta::T1
533533
beta::T2
534-
epsilon::T3
535534
lambda::T4
535+
epsilon::T3
536536
couple::Bool
537537
end
538538

539539
function AdamW(η, β = (0.9, 0.999), λ = 0.0, ϵ = 1e-8; couple::Bool = true)
540540
η < 0 && throw(DomainError(η, "the learning rate cannot be negative"))
541-
AdamW(η, β, ϵ, λ, couple)
541+
AdamW(η, β, λ, ϵ, couple)
542542
end
543543

544544
AdamW(; eta = 0.001, beta = (0.9, 0.999), lambda= 0.0, epsilon = 1e-8, kw...) =

0 commit comments

Comments
 (0)