From 99a7697d13e6b50c29b7fd6739e39ea887727338 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 18 Oct 2017 17:44:21 +0100 Subject: [PATCH] adam eta default arg --- src/optimise/interface.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/optimise/interface.jl b/src/optimise/interface.jl index 8e63bb26..0b2a25ae 100644 --- a/src/optimise/interface.jl +++ b/src/optimise/interface.jl @@ -51,7 +51,7 @@ RMSProp(ps, η = 0.001; ρ = 0.9, ϵ = 1e-8, decay = 0) = [ADAM](https://arxiv.org/abs/1412.6980v8) optimiser. """ -ADAM(ps; η = 0.001, β1 = 0.9, β2 = 0.999, ϵ = 1e-08, decay = 0) = +ADAM(ps, η = 0.001; β1 = 0.9, β2 = 0.999, ϵ = 1e-08, decay = 0) = optimiser(ps, p -> adam(p; η = η, β1 = β1, β2 = β2, ϵ = ϵ), p -> invdecay(p, decay), p -> descent(p, 1)) """