Momentum doesn't need params
This commit is contained in:
parent
bab618d168
commit
ed12d4e7c0
@ -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 `ρ`.
|
||||
"""
|
||||
|
Loading…
Reference in New Issue
Block a user