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

fix param interpretation

Browse Source
This commit is contained in:
Mike J Innes 2016-12-13 15:46:34 +00:00
parent a63cd826c2
commit 1b22d55401
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/backend/tensorflow/graph.jl
View File

@ -42,6 +42,8 @@ interp{T<:AArray}(ctx, p::Constant{Flux.Param{T}}) =
ctx[:params][p.value] : ctx[:params][p.value] :
(ctx[:params][p.value] = Variable(p.value.x)) (ctx[:params][p.value] = Variable(p.value.x))
interp(ctx, p::Constant) = p.value
function interp(ctx, model, args...) function interp(ctx, model, args...)
g = Flux.graph(model) g = Flux.graph(model)
g == nothing && return graph(model, interpret(ctx, args)...) g == nothing && return graph(model, interpret(ctx, args)...)
@ -50,7 +52,7 @@ function interp(ctx, model, args...)
end end
function tograph(model, args...) function tograph(model, args...)
ctx = Context(interplambda(interptuple(interpmap(interpconst(interp)))), params = ObjectIdDict()) ctx = Context(interplambda(interptuple(interpmap(interp))), params = ObjectIdDict())
out = interp(ctx, model, map(constant, args)...) out = interp(ctx, model, map(constant, args)...)
return ctx[:params], out return ctx[:params], out
end end