From db99e41959ba943ec5c7a0e4d561cf97690c2637 Mon Sep 17 00:00:00 2001 From: Bruno Hebling Vieira Date: Thu, 16 Apr 2020 09:50:41 -0300 Subject: [PATCH] Removed SGD exports --- src/Flux.jl | 2 +- src/optimise/Optimise.jl | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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