diff --git a/src/rules.jl b/src/rules.jl index 0063d70..47c2d8c 100644 --- 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