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

deprecation

Browse Source
This commit is contained in:
Mike J Innes 2016-08-22 14:55:12 +01:00
parent 5995597ebc
commit 0496ce6bda
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/compiler/code.jl
View File

@ -24,7 +24,7 @@ function build_type(T, params)
quote quote
type $T <: Model type $T <: Model
$(params...) $(params...)
$([symbol("Δ", s) for s in params]...) $([Symbol("Δ", s) for s in params]...)
end end
$T($(params...)) = $T($(params...), $T($(params...)) = $T($(params...),
$((:(zeros($p)) for p in params)...)) $((:(zeros($p)) for p in params)...))
@ -40,7 +40,7 @@ function build_backward(body, x, params)
back = IVertex{Any}(Flow.Do()) back = IVertex{Any}(Flow.Do())
for param in params for param in params
haskey(Δs, :(self.$param)) || continue haskey(Δs, :(self.$param)) || continue
k = symbol("Δ", param) k = Symbol("Δ", param)
ksym = Expr(:quote, k) ksym = Expr(:quote, k)
ex = Δs[:(self.$param)] ex = Δs[:(self.$param)]
thread!(back, @dvertex(setfield!(:self, ksym, :(self.$k) + ex))) thread!(back, @dvertex(setfield!(:self, ksym, :(self.$k) + ex)))
@ -53,7 +53,7 @@ end
function build_update(T, params) function build_update(T, params)
updates = [] updates = []
for p in params for p in params
Δp = symbol("Δ", p) Δp = Symbol("Δ", p)
push!(updates, :(self.$p += self.$Δp; fill!(self.$Δp, 0))) push!(updates, :(self.$p += self.$Δp; fill!(self.$Δp, 0)))
end end
:(update!(self::$T) = $(updates...)) :(update!(self::$T) = $(updates...))