returning

2018-08-18 08:24:49 +05:30 · 2018-08-18 08:24:49 +05:30 · 65a5ecccd2
commit 65a5ecccd2
parent 999b00b64d
1 changed files with 2 additions and 2 deletions
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@ -37,9 +37,9 @@ function train!(loss, data, opt; cb = () -> ())
  cb = try
        runall(cb)
      catch e
-        if e isa StopException || rethrow()
+        if e isa StopException || rethrow(e)
          @info "Stop Condition Met"
-          break
+          :stop
  opt = runall(opt)
  @progress for d in data
    l = loss(d...)