Flux.jl/src/optimise/optimisers.jl

15 lines
194 B
Julia
Raw Normal View History

2017-08-22 21:25:18 +00:00
struct SGD
ps::Vector{Any}
η::Float32
end
sgd(m, η) = SGD(params(m), η)
function update!(o::SGD)
for p in o.ps
x, Δ = data(p), grad(p)
x .-= Δ .* o.η
2017-08-24 10:42:29 +00:00
Δ .= 0
2017-08-22 21:25:18 +00:00
end
end