2017-09-01 21:06:51 +00:00
|
|
|
call(f, xs...) = f(xs...)
|
|
|
|
|
|
|
|
|
|
function optimiser(m, fs...)
|
|
|
|
|
ps = Param.(params(m))
|
|
|
|
|
fs = map(ps) do p
|
|
|
|
|
os = map(f -> f(p), fs)
|
|
|
|
|
() -> foreach(call, os)
|
|
|
|
|
end
|
|
|
|
|
() -> foreach(call, fs)
|
|
|
|
|
end
|
|
|
|
|
|
2017-09-02 03:41:44 +00:00
|
|
|
SGD(m, η = 1) = optimiser(m, p -> descent(p, η))
|
