From 887bfad312467c67e3ba894f771baf72c15c87cb Mon Sep 17 00:00:00 2001
From: Dhairya Gandhi <dhairya@juliacopmuting.com>
Date: Sat, 18 Aug 2018 08:28:47 +0530
Subject: [PATCH] returning :stop

---
 src/optimise/train.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/optimise/train.jl b/src/optimise/train.jl
index c7899b08..3bf06d8f 100644
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@@ -39,7 +39,7 @@ function train!(loss, data, opt; cb = () -> ())
       catch e
         if e isa StopException || rethrow(e)
           @info "Stop Condition Met"
-          :stop
+          return :stop
   opt = runall(opt)
   @progress for d in data
     l = loss(d...)