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

6
src/Flux.jl
View File

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

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(η)