diff --git a/src/optimise/Optimise.jl b/src/optimise/Optimise.jl index 4ea5235e..0f5e644f 100644 --- a/src/optimise/Optimise.jl +++ b/src/optimise/Optimise.jl @@ -2,10 +2,11 @@ module Optimise using LinearAlgebra -export train!, update!, stop, Optimiser, - Descent, ADAM, Momentum, Nesterov, RMSProp, - ADAGrad, AdaMax, ADADelta, AMSGrad, NADAM, ADAMW, RADAM, - InvDecay, ExpDecay, WeightDecay, ClipValue, ClipNorm +export train!, update!, + Descent, ADAM, Momentum, Nesterov, RMSProp, + ADAGrad, AdaMax, ADADelta, AMSGrad, NADAM, ADAMW,RADAM, + InvDecay, ExpDecay, WeightDecay, stop, Optimiser, + ClipValue, ClipNorm include("optimisers.jl") include("train.jl")