diff --git a/src/Flux.jl b/src/Flux.jl index 9969b323..e453e129 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -16,7 +16,7 @@ export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, CrossCor, ConvTranspose, MaxP include("optimise/Optimise.jl") using .Optimise using .Optimise: @epochs -export SGD, Descent, ADAM, Momentum, Nesterov, RMSProp, +export Descent, ADAM, Momentum, Nesterov, RMSProp, ADAGrad, AdaMax, ADADelta, AMSGrad, NADAM, ADAMW, RADAM, InvDecay, ExpDecay, WeightDecay diff --git a/src/optimise/Optimise.jl b/src/optimise/Optimise.jl index 68c18a6f..f25e9a50 100644 --- a/src/optimise/Optimise.jl +++ b/src/optimise/Optimise.jl @@ -1,7 +1,7 @@ module Optimise export train!, - SGD, Descent, ADAM, Momentum, Nesterov, RMSProp, + Descent, ADAM, Momentum, Nesterov, RMSProp, ADAGrad, AdaMax, ADADelta, AMSGrad, NADAM, ADAMW,RADAM, InvDecay, ExpDecay, WeightDecay, stop, Optimiser