diff --git a/src/optimise/Optimise.jl b/src/optimise/Optimise.jl index ee7723bc..c4828c9e 100644 --- a/src/optimise/Optimise.jl +++ b/src/optimise/Optimise.jl @@ -2,7 +2,7 @@ module Optimise export train!, SGD, ADAM, ADAMW, AdaMax, Momentum, Nesterov, - RMSProp, ADAGrad, ADADelta, AMSGrad, NADAM + RMSProp, ADAGrad, ADADelta, AMSGrad, NADAM, stop, StopException struct Param{T} x::T