diff --git a/src/optimise/Optimise.jl b/src/optimise/Optimise.jl index 57956426..4a30c32f 100644 --- a/src/optimise/Optimise.jl +++ b/src/optimise/Optimise.jl @@ -1,7 +1,7 @@ module Optimise export update!, params, train!, - SGD + SGD, ADAM, Momentum, Nesterov, RMSProp, ADAGrad, ADADelta struct Param{T} x::T