fix for new api

2016-08-24 12:11:34 +01:00 · 2016-08-24 12:11:34 +01:00 · 8224c77f7d
parent 5ea18ddffd
commit 8224c77f7d
1 changed files with 1 additions and 1 deletions
--- 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)