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

tweaks

Browse Source
This commit is contained in:
Mike J Innes 2016-08-23 15:09:15 +01:00
parent 20b13b9a41
commit 545d4480ed
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

@ -6,7 +6,6 @@ end
param(x) = Param(x, zero(x)) param(x) = Param(x, zero(x))
state(p::Param) = p.x state(p::Param) = p.x
state(x) = x
function accumulate!(p::Param, Δ) function accumulate!(p::Param, Δ)
p.Δx .+= Δ p.Δx .+= Δ
@ -14,8 +13,9 @@ function accumulate!(p::Param, Δ)
end end
function update!(p::Param, η) function update!(p::Param, η)
p.x .+= p.Δx * η p.x .+= p.Δx .* η
return p return p
end end
state(x) = x
accumulate!(x, Δ) = x accumulate!(x, Δ) = x