cb convenience

2017-09-07 00:27:16 -04:00 · 2017-09-07 00:27:16 -04:00 · aeaa138b6d
commit aeaa138b6d
parent e837bb0745
1 changed files with 4 additions and 0 deletions
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@ -1,7 +1,11 @@
 using Juno
 using Flux.Tracker: back!

+tocb(f) = f
+tocb(fs::AbstractVector) = () -> foreach(call, fs)
+
 function train!(m, data, opt; epoch = 1, cb = () -> ())
+  cb = tocb(cb)
  @progress for e in 1:epoch
    epoch > 1 && info("Epoch $e")
    @progress for (x, y) in data