From 669798c19ec5d629769ee0b0bf1c0509d2ec157b Mon Sep 17 00:00:00 2001
From: Carlo Lucibello <carlo.lucibello@gmail.com>
Date: Wed, 11 Dec 2024 16:54:42 +0100
Subject: [PATCH] fix AdamW (#198)

---
 src/rules.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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