From 3f83be7bb7cdcdf56ec19514d0639ae85d6c9c56 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Mon, 11 Sep 2017 13:11:55 +0100 Subject: [PATCH] more flexible training loop --- src/optimise/train.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/optimise/train.jl b/src/optimise/train.jl index 02de7b21..29916b13 100644 --- 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