Merge pull request #44 from darsnack/optimiser-adam-deprecations

Update optimiser names with deprecations
FluxML · Feb 17, 2023 · 715b92a · 715b92a · darsnack · Feb 18, 2023
2 parents c722a40 + fd9ab6c
commit 715b92a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "ParameterSchedulers"
 uuid = "d7d3b36b-41b8-4d0d-a2bf-768c6151755e"
 authors = ["Kyle Daruwalla"]
-version = "0.3.4"
+version = "0.3.5"
 
 [deps]
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"

diff --git a/src/ParameterSchedulers.jl b/src/ParameterSchedulers.jl
@@ -80,8 +80,8 @@ function Flux.Optimise.apply!(opt::Scheduler, x, Δ)
 end
 
 for Opt in (Descent, Momentum, Nesterov, RMSProp,
-            ADAM, RADAM, AdaMax, OADAM, ADAGrad,
-            ADADelta, AMSGrad, NADAM, AdaBelief)
+            Adam, RAdam, AdaMax, OAdam, AdaGrad,
+            AdaDelta, AMSGrad, NAdam, AdaBelief)
     @eval begin
         Scheduler(schedule, opt::$Opt; update_func = (o, s) -> (o.eta = s)) =
             Scheduler(schedule, opt, update_func)