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

more flexible training loop

Browse Source
This commit is contained in:
Mike J Innes 2017-09-11 13:11:55 +01:00
parent c2db42d38e
commit 3f83be7bb7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/optimise/train.jl
View File

@ -6,8 +6,8 @@ tocb(fs::AbstractVector) = () -> foreach(call, fs)
function train!(m, data, opt; cb = () -> ()) function train!(m, data, opt; cb = () -> ())
cb = tocb(cb) cb = tocb(cb)
@progress for (x, y) in data @progress for x in data
back!(m(x, y)) back!(m(x...))
opt() opt()
cb() cb()
end end