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

juno progress

Browse Source
This commit is contained in:
Mike J Innes 2017-09-03 02:44:32 -04:00
parent bd5822fd71
commit e57ae77bbb
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/optimise/train.jl
View File

@ -1,9 +1,10 @@
using Juno
using Flux.Tracker: back! using Flux.Tracker: back!
function train!(m, data, opt; epoch = 1, cb = () -> ()) function train!(m, data, opt; epoch = 1, cb = () -> ())
for e in 1:epoch @progress for e in 1:epoch
epoch > 1 && info("Epoch $e") epoch > 1 && info("Epoch $e")
for (x, y) in data @progress for (x, y) in data
back!(m(x, y)) back!(m(x, y))
opt() opt()
cb() cb()