From ed12d4e7c04207a44e4c11a96b970228fd3b16e1 Mon Sep 17 00:00:00 2001 From: Christopher Rackauckas Date: Wed, 31 Jul 2019 17:56:51 -0400 Subject: [PATCH] Momentum doesn't need params --- src/optimise/optimisers.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/optimise/optimisers.jl b/src/optimise/optimisers.jl index 2319cfdb..939a4678 100644 --- a/src/optimise/optimisers.jl +++ b/src/optimise/optimisers.jl @@ -23,7 +23,7 @@ function apply!(o::Descent, x, Δ) end """ - Momentum(params, η = 0.01; ρ = 0.9) + Momentum(η = 0.01; ρ = 0.9) Gradient descent with learning rate `η` and momentum `ρ`. """