2017-08-22 16:13:03 +00:00
|
|
|
module Optimise
|
|
|
|
|
|
2017-09-01 21:06:51 +00:00
|
|
|
export update!, params, train!,
|
|
|
|
|
SGD
|
2017-08-22 21:25:18 +00:00
|
|
|
|
2017-08-22 16:13:03 +00:00
|
|
|
include("params.jl")
|
2017-08-22 21:25:18 +00:00
|
|
|
include("optimisers.jl")
|
2017-09-01 21:06:51 +00:00
|
|
|
include("interface.jl")
|
2017-08-24 10:42:29 +00:00
|
|
|
include("train.jl")
|
2017-08-22 16:13:03 +00:00
|
|
|
|
2017-08-31 16:36:18 +00:00
|
|
|
using Flux.Tracker: TrackedArray
|
|
|
|
|
|
|
|
|
|
params(ps, p::TrackedArray) = push!(ps, p)
|
|
|
|
|
|
2017-09-07 03:09:32 +00:00
|
|
|
Base.convert(::Type{Param}, x::TrackedArray) = Param(x.data, x.grad[])
|
2017-08-31 16:36:18 +00:00
|
|
|
|
2017-08-22 16:13:03 +00:00
|
|
|
end
|
