From b194e7e3a898ac8425841e6246421b8fac3c879b Mon Sep 17 00:00:00 2001
From: thebhatman <manjunathbhat9920@gmail.com>
Date: Thu, 20 Jun 2019 00:37:54 +0530
Subject: [PATCH] Callback being called now

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

diff --git a/src/optimise/train.jl b/src/optimise/train.jl
index 6317b3ec..07577e94 100644
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@@ -67,6 +67,7 @@ function train!(loss, ps, data, opt; cb = () -> ())
         loss(d...)
       end
       update!(opt, ps, gs)
+      cb()
     catch ex
       if ex isa StopException
         break