Flux.jl/src/optimise/train.jl

using Juno
using Flux.Tracker: back!

function train!(m, data, opt; epoch = 1, cb = () -> ())
  @progress for e in 1:epoch
    epoch > 1 && info("Epoch $e")
    @progress for (x, y) in data
      back!(m(x, y))
      opt()
      cb()
    end
  end
end
juno progress 2017-09-03 06:44:32 +00:00			`using Juno`
opt refactor 2017-08-31 16:36:18 +00:00			`using Flux.Tracker: back!`

add callbacks back 2017-09-02 03:59:44 +00:00			`function train!(m, data, opt; epoch = 1, cb = () -> ())`
juno progress 2017-09-03 06:44:32 +00:00			`@progress for e in 1:epoch`
basic training loop 2017-08-24 10:42:29 +00:00			`epoch > 1 && info("Epoch $e")`
juno progress 2017-09-03 06:44:32 +00:00			`@progress for (x, y) in data`
add callbacks back 2017-09-02 03:59:44 +00:00			`back!(m(x, y))`
optimisers rework 2017-09-01 21:06:51 +00:00			`opt()`
add callbacks back 2017-09-02 03:59:44 +00:00			`cb()`
basic training loop 2017-08-24 10:42:29 +00:00			`end`
			`end`
			`end`