TastyPancakes
/
Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix for new api

This commit is contained in:
Mike J Innes 2016-08-24 12:11:34 +01:00
parent 5ea18ddffd
commit 8224c77f7d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/utils.jl
View File

@ -12,7 +12,7 @@ function train!(m::Model, train, test = []; epoch = 1, batch = 10, η = 0.1)
for (x, y) in shuffle!(train) for (x, y) in shuffle!(train)
i += 1 i += 1
err = mse!(∇, m(x), y) err = mse!(∇, m(x), y)
back!(m, ∇) back!(m, ∇, x)
i % batch == 0 && update!(m, η/batch) i % batch == 0 && update!(m, η/batch)
end end
@show accuracy(m, test) @show accuracy(m, test)