From 8224c77f7daafd981ff8456da0c047c1ccc41e2a Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Wed, 24 Aug 2016 12:11:34 +0100
Subject: [PATCH] fix for new api

---
 src/utils.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/utils.jl b/src/utils.jl
index b02bab53..5f81462e 100644
--- a/src/utils.jl
+++ b/src/utils.jl
@@ -12,7 +12,7 @@ function train!(m::Model, train, test = []; epoch = 1, batch = 10, η = 0.1)
       for (x, y) in shuffle!(train)
         i += 1
         err = mse!(∇, m(x), y)
-        back!(m, ∇)
+        back!(m, ∇, x)
         i % batch == 0 && update!(m, η/batch)
       end
       @show accuracy(m, test)