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

add gradients in place

Browse Source
This commit is contained in:
Mike J Innes 2016-08-23 15:03:06 +01:00
parent ddce04ea9d
commit c4b96dbc98
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/layers/params.jl
View File

@ -9,12 +9,12 @@ state(p::Param) = p
state(x) = x state(x) = x
function accumulate!(p::Param, Δ) function accumulate!(p::Param, Δ)
p.Δx += Δ p.Δx .+= Δ
return p return p
end end
function update!(p::Param, η) function update!(p::Param, η)
p.x += p.Δx * η p.x .+= p.Δx * η
return p return p
end end