Skip to content

Commit 051270b

Browse files
Merge pull request #8 from FluxML/dg/rename
Remove noarg constructors for some optimisers
2 parents 18a4939 + 21dc5b8 commit 051270b

File tree

2 files changed

+1
-6
lines changed

2 files changed

+1
-6
lines changed

src/rules.jl

-5
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,6 @@ struct Momentum{T,S}
2828
rho::S
2929
end
3030

31-
Momentum= 0.01, ρ = 0.9) = Momentum{typeof(η), typeof(ρ)}(η, ρ)
32-
3331
function apply(o::Momentum, x, Δ, st)
3432
η, ρ = o.eta, o.rho
3533
v = st
@@ -49,8 +47,6 @@ struct Nesterov{T,S}
4947
rho::S
5048
end
5149

52-
Nesterov= 0.001, ρ = 0.9) = Nesterov{typeof(η), typeof(ρ)}(η, ρ)
53-
5450
init(o::Nesterov, x::AbstractArray) = zero(x)
5551

5652
function (o::Nesterov)(m, m̄, state)
@@ -72,7 +68,6 @@ struct RMSProp{T,S}
7268
end
7369

7470
init(o::RMSProp, x::AbstractArray) = zero(x)
75-
RMSProp= 0.001, ρ = 0.9) = RMSProp{typeof(η), typeof(ρ)}(η, ρ)
7671

7772
function apply(o::RMSProp, x, Δ, st)
7873
η, ρ = o.eta, o.rho

test/runtests.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ using Statistics
55
@testset "Optimisers" begin
66
Random.seed!(84)
77
w′ = rand(3,3)
8-
@testset for o in (Descent(0.1), Momentum(), Nesterov(), RMSProp(),
8+
@testset for o in (Descent(0.1), Momentum(0.01, 0.9), Nesterov(0.001, 0.9), RMSProp(0.001, 0.9),
99
ADAM(0.001, (0.9, 0.99)))
1010
w = rand(3,3)
1111
st = Optimisers.init(o,w)

0 commit comments

Comments
 (0)