Momentum doesn't need params

This commit is contained in:
Christopher Rackauckas 2019-07-31 17:56:51 -04:00 committed by GitHub
parent bab618d168
commit ed12d4e7c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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