more flexible training loop

2017-09-11 13:11:55 +01:00 · 2017-09-11 13:11:55 +01:00 · 3f83be7bb7
commit 3f83be7bb7
parent c2db42d38e
1 changed files with 2 additions and 2 deletions
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@ -6,8 +6,8 @@ tocb(fs::AbstractVector) = () -> foreach(call, fs)

 function train!(m, data, opt; cb = () -> ())
  cb = tocb(cb)
-  @progress for (x, y) in data
-    back!(m(x, y))
+  @progress for x in data
+    back!(m(x...))
    opt()
    cb()
  end