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 `ρ`. """