TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

correct SGD deprecation

Browse Source
This commit is contained in:
Mike J Innes 2018-10-31 15:30:30 +00:00
parent bffaceee02
commit 4a54d30cbf
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Flux.jl
View File

@ -19,7 +19,7 @@ export Tracker, TrackedArray, TrackedVector, TrackedMatrix, param
include("optimise/Optimise.jl")
using .Optimise
using .Optimise: @epochs
export Descent, ADAM, Momentum, Nesterov, RMSProp,
export SGD, Descent, ADAM, Momentum, Nesterov, RMSProp,
ADAGrad, AdaMax, ADADelta, AMSGrad, NADAM,
ADAMW, InvDecay, ExpDecay, WeightDecay

2
src/optimise/Optimise.jl
View File

@ -1,7 +1,7 @@
module Optimise
export train!,
Descent, ADAM, Momentum, Nesterov, RMSProp,
SGD, Descent, ADAM, Momentum, Nesterov, RMSProp,
ADAGrad, AdaMax, ADADelta, AMSGrad, NADAM, ADAMW,
InvDecay, ExpDecay, WeightDecay, stop, Optimiser

4
src/optimise/deprecations.jl
View File

@ -12,8 +12,8 @@ function updaterule(opt, ps)
end
end
function Descent(params::AbstractArray, η = 0.1; decay = 0.)
depwarn("Descent(params) is deprecated; use Descent(η::Float64) instead", :Descent)
function SGD(params::AbstractArray, η = 0.1; decay = 0.)
depwarn("SGD(params) is deprecated; use Descent(η::Float64) instead", :SGD)
ps = params
opt = Descent(η)