2017-08-31 16:36:18 +00:00
|
|
|
using Flux.Tracker: back!
|
|
|
|
|
|
2017-08-24 10:42:29 +00:00
|
|
|
function train!(m, data, opt; epoch = 1)
|
|
|
|
|
for e in 1:epoch
|
|
|
|
|
epoch > 1 && info("Epoch $e")
|
|
|
|
|
for (x, y) in data
|
|
|
|
|
loss = m(x, y)
|
|
|
|
|
@show loss
|
|
|
|
|
back!(loss)
|
2017-09-01 21:06:51 +00:00
|
|
|
opt()
|
2017-08-24 10:42:29 +00:00
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
