moving stop to Optimise

src/Flux.jl

@@ -7,7 +7,7 @@ using MacroTools: @forward
 
 export Chain, Dense, RNN, LSTM, GRU, Conv,
        Dropout, LayerNorm, BatchNorm,
-       params, mapleaves, cpu, gpu, stop, StopException
+       params, mapleaves, cpu, gpu
 
 @reexport using NNlib
 using NNlib: @fix
@@ -21,7 +21,7 @@ include("optimise/Optimise.jl")
 using .Optimise
 using .Optimise: @epochs
 export SGD, ADAM, ADAMW, AdaMax, Momentum, Nesterov,
-       RMSProp, ADAGrad, ADADelta, AMSGrad, NADAM
+       RMSProp, ADAGrad, ADADelta, AMSGrad, NADAM, stop, StopException
 
 include("utils.jl")
 include("onehot.jl")

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

src/optimise/train.jl

@@ -1,6 +1,5 @@
 using Juno
 using Flux.Tracker: back!
-using Flux: stop, StopException
 
 runall(f) = f
 runall(fs::AbstractVector) = () -> foreach(call, fs)
@@ -15,6 +14,13 @@ macro interrupts(ex)
     end)
 end
 
+struct StopException <: Exception 
+  x::Symbol
+end
+
+function stop(x)
+  throw(StopException(
+
 """
     train!(loss, data, opt)
 

src/utils.jl

@@ -154,8 +154,10 @@ function jacobian(m,x)
     J'
 end
 
-struct StopException <: Exception end
+# struct StopException <: Exception 
+#   x::Symbol
+# end
 
-function stop()
+# function stop(x)
-  throw(StopException())
+#   throw(StopException(x))
-end
+# end