Skip to content

Commit 12a01c5

Browse files
author
cossio
committed
move eps to the end
1 parent 6cd2bf7 commit 12a01c5

File tree

1 file changed

+20
-20
lines changed

1 file changed

+20
-20
lines changed

src/optimise/optimisers.jl

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -133,11 +133,11 @@ opt = RMSProp(0.002, 0.95)
133133
mutable struct RMSProp <: AbstractOptimiser
134134
eta::Float64
135135
rho::Float64
136-
epsilon::Float64
137136
acc::IdDict
137+
epsilon::Float64
138138
end
139139

140-
RMSProp= 0.001, ρ = 0.9, ϵ = ϵ) = RMSProp(η, ρ, ϵ, IdDict())
140+
RMSProp= 0.001, ρ = 0.9; ϵ = ϵ) = RMSProp(η, ρ, IdDict(), ϵ)
141141

142142
function apply!(o::RMSProp, x, Δ)
143143
η, ρ = o.eta, o.rho
@@ -167,11 +167,11 @@ opt = ADAM(0.001, (0.9, 0.8))
167167
mutable struct ADAM <: AbstractOptimiser
168168
eta::Float64
169169
beta::Tuple{Float64,Float64}
170-
epsilon::Float64
171170
state::IdDict
171+
epsilon::Float64
172172
end
173173

174-
ADAM= 0.001, β = (0.9, 0.999), ϵ = ϵ) = ADAM(η, β, ϵ, IdDict())
174+
ADAM= 0.001, β = (0.9, 0.999); ϵ = ϵ) = ADAM(η, β, IdDict(), ϵ)
175175

176176
function apply!(o::ADAM, x, Δ)
177177
η, β = o.eta, o.beta
@@ -209,11 +209,11 @@ opt = RADAM(0.001, (0.9, 0.8))
209209
mutable struct RADAM <: AbstractOptimiser
210210
eta::Float64
211211
beta::Tuple{Float64,Float64}
212-
epsilon::Float64
213212
state::IdDict
213+
epsilon::Float64
214214
end
215215

216-
RADAM= 0.001, β = (0.9, 0.999), ϵ = ϵ) = RADAM(η, β, ϵ, IdDict())
216+
RADAM= 0.001, β = (0.9, 0.999); ϵ = ϵ) = RADAM(η, β, IdDict(), ϵ)
217217

218218
function apply!(o::RADAM, x, Δ)
219219
η, β = o.eta, o.beta
@@ -259,11 +259,11 @@ opt = AdaMax(0.001, (0.9, 0.995))
259259
mutable struct AdaMax <: AbstractOptimiser
260260
eta::Float64
261261
beta::Tuple{Float64,Float64}
262-
epsilon::Float64
263262
state::IdDict
263+
epsilon::Float64
264264
end
265265

266-
AdaMax= 0.001, β = (0.9, 0.999), ϵ = ϵ) = AdaMax(η, β, ϵ, IdDict())
266+
AdaMax= 0.001, β = (0.9, 0.999); ϵ = ϵ) = AdaMax(η, β, IdDict(), ϵ)
267267

268268
function apply!(o::AdaMax, x, Δ)
269269
η, β = o.eta, o.beta
@@ -302,11 +302,11 @@ opt = OADAM(0.001, (0.9, 0.995))
302302
mutable struct OADAM <: AbstractOptimiser
303303
eta::Float64
304304
beta::Tuple{Float64,Float64}
305-
epsilon::Float64
306305
state::IdDict
306+
epsilon::Float64
307307
end
308308

309-
OADAM= 0.001, β = (0.5, 0.9), ϵ = ϵ) = OADAM(η, β, ϵ, IdDict())
309+
OADAM= 0.001, β = (0.5, 0.9); ϵ = ϵ) = OADAM(η, β, IdDict(), ϵ)
310310

311311
function apply!(o::OADAM, x, Δ)
312312
η, β = o.eta, o.beta
@@ -345,11 +345,11 @@ opt = ADAGrad(0.001)
345345
"""
346346
mutable struct ADAGrad <: AbstractOptimiser
347347
eta::Float64
348-
epsilon::Float64
349348
acc::IdDict
349+
epsilon::Float64
350350
end
351351

352-
ADAGrad= 0.1, ϵ = ϵ) = ADAGrad(η, ϵ, IdDict())
352+
ADAGrad= 0.1; ϵ = ϵ) = ADAGrad(η, IdDict(), ϵ)
353353

354354
function apply!(o::ADAGrad, x, Δ)
355355
η = o.eta
@@ -377,11 +377,11 @@ opt = ADADelta(0.89)
377377
"""
378378
mutable struct ADADelta <: AbstractOptimiser
379379
rho::Float64
380-
epsilon::Float64
381380
state::IdDict
381+
epsilon::Float64
382382
end
383383

384-
ADADelta= 0.9, ϵ = ϵ) = ADADelta(ρ, ϵ, IdDict())
384+
ADADelta= 0.9; ϵ = ϵ) = ADADelta(ρ, IdDict(), ϵ)
385385

386386
function apply!(o::ADADelta, x, Δ)
387387
ρ = o.rho
@@ -416,11 +416,11 @@ opt = AMSGrad(0.001, (0.89, 0.995))
416416
mutable struct AMSGrad <: AbstractOptimiser
417417
eta::Float64
418418
beta::Tuple{Float64, Float64}
419-
epsilon::Float64
420419
state::IdDict
420+
epsilon::Float64
421421
end
422422

423-
AMSGrad= 0.001, β = (0.9, 0.999), ϵ = ϵ) = AMSGrad(η, β, ϵ, IdDict())
423+
AMSGrad= 0.001, β = (0.9, 0.999); ϵ = ϵ) = AMSGrad(η, β, IdDict(), ϵ)
424424

425425
function apply!(o::AMSGrad, x, Δ)
426426
η, β = o.eta, o.beta
@@ -457,11 +457,11 @@ opt = NADAM(0.002, (0.89, 0.995))
457457
mutable struct NADAM <: AbstractOptimiser
458458
eta::Float64
459459
beta::Tuple{Float64, Float64}
460-
epsilon::Float64
461460
state::IdDict
461+
epsilon::Float64
462462
end
463463

464-
NADAM= 0.001, β = (0.9, 0.999), ϵ = ϵ) = NADAM(η, β, ϵ, IdDict())
464+
NADAM= 0.001, β = (0.9, 0.999); ϵ = ϵ) = NADAM(η, β, IdDict(), ϵ)
465465

466466
function apply!(o::NADAM, x, Δ)
467467
η, β = o.eta, o.beta
@@ -524,11 +524,11 @@ opt = AdaBelief(0.001, (0.9, 0.8))
524524
mutable struct AdaBelief
525525
eta::Float64
526526
beta::Tuple{Float64,Float64}
527-
epsilon::Float64
528527
state::IdDict
528+
epsilon::Float64
529529
end
530530

531-
AdaBelief= 0.001, β = (0.9, 0.999), ϵ = ϵ) = AdaBelief(η, β, ϵ, IdDict())
531+
AdaBelief= 0.001, β = (0.9, 0.999); ϵ = ϵ) = AdaBelief(η, β, IdDict(), ϵ)
532532

533533
function apply!(o::AdaBelief, x, Δ)
534534
η, β = o.eta, o.beta

0 commit comments

Comments
 (0)