Flux.jl/src/optimise/interface.jl

13 lines
253 B
Julia
Raw Normal View History

2017-09-01 21:06:51 +00:00
call(f, xs...) = f(xs...)
2017-09-03 21:10:04 +00:00
function optimiser(ps, fs...)
ps = [Param(p) for p in ps]
2017-09-01 21:06:51 +00:00
fs = map(ps) do p
os = map(f -> f(p), fs)
() -> foreach(call, os)
end
() -> foreach(call, fs)
end
2017-09-03 21:10:04 +00:00
SGD(ps, η = 1) = optimiser(ps, p -> descent(p, η))