Callback being called now

2019-06-20 00:37:54 +05:30 · 2019-06-20 00:37:54 +05:30 · b194e7e3a8
commit b194e7e3a8
parent e6d5846e49
1 changed files with 1 additions and 0 deletions
--- 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