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