2017-08-22 16:13:03 +00:00
|
|
|
module Optimise
|
|
|
|
|
|
2017-08-24 21:23:05 +00:00
|
|
|
using ..Tracker: TrackedArray, data, grad, back!
|
2017-08-22 21:25:18 +00:00
|
|
|
|
2017-08-24 10:42:29 +00:00
|
|
|
export sgd, update!, params, train!
|
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-08-24 10:42:29 +00:00
|
|
|
include("train.jl")
|
2017-08-22 16:13:03 +00:00
|
|
|
|
|
|
|
|
end
|
