TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tweak optimiser interface

Browse Source
This commit is contained in:
Mike J Innes 2017-09-03 17:10:04 -04:00
parent d4211b1f23
commit 47ba702747
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/optimise/interface.jl
View File

@ -1,7 +1,7 @@
call(f, xs...) = f(xs...) call(f, xs...) = f(xs...)
function optimiser(m, fs...) function optimiser(ps, fs...)
ps = Param.(params(m)) ps = [Param(p) for p in ps]
fs = map(ps) do p fs = map(ps) do p
os = map(f -> f(p), fs) os = map(f -> f(p), fs)
() -> foreach(call, os) () -> foreach(call, os)
@ -9,4 +9,4 @@ function optimiser(m, fs...)
() -> foreach(call, fs) () -> foreach(call, fs)
end end
SGD(m, η = 1) = optimiser(m, p -> descent(p, η)) SGD(ps, η = 1) = optimiser(ps, p -> descent(p, η))