Flux.jl/src/optimise/train.jl

using Flux.Tracker: back!

function train!(m, data, opt; epoch = 1, cb = () -> ())
  for e in 1:epoch
    epoch > 1 && info("Epoch $e")
    for (x, y) in data
      back!(m(x, y))
      opt()
      cb()
    end
  end
end
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 = () -> ())`
basic training loop 2017-08-24 10:42:29 +00:00			`for e in 1:epoch`
			`epoch > 1 && info("Epoch $e")`
			`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`