From 65a5ecccd2823d011f5ba7a0375fe465327e4fa4 Mon Sep 17 00:00:00 2001
From: Dhairya Gandhi <dhairya@juliacopmuting.com>
Date: Sat, 18 Aug 2018 08:24:49 +0530
Subject: [PATCH] returning

---
 src/optimise/train.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/optimise/train.jl b/src/optimise/train.jl
index a3af8d66..c7899b08 100644
--- 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...)