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

add callbacks back

Browse Source
This commit is contained in:
Mike J Innes 2017-09-01 23:59:44 -04:00
parent bf098d551c
commit fe2b35facc
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/optimise/train.jl
View File

@ -1,13 +1,12 @@
using Flux.Tracker: back! using Flux.Tracker: back!
function train!(m, data, opt; epoch = 1) function train!(m, data, opt; epoch = 1, cb = () -> ())
for e in 1:epoch for e in 1:epoch
epoch > 1 && info("Epoch $e") epoch > 1 && info("Epoch $e")
for (x, y) in data for (x, y) in data
loss = m(x, y) back!(m(x, y))
@show loss
back!(loss)
opt() opt()
cb()
end end
end end
end end