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

style tweak

Browse Source
This commit is contained in:
Mike J Innes 2017-07-27 20:43:53 +01:00
parent dd82edd818
commit 3000c7bbcb
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/backend/tensorflow/model.jl
View File

@ -13,9 +13,11 @@ function makesession(model, inputs; session = Session(Graph()))
params, stacks, output = tograph(model, inputs...) params, stacks, output = tograph(model, inputs...)
output = mapt(x->Param{Tensor}(x, placeholder(Float32)), output) output = mapt(x->Param{Tensor}(x, placeholder(Float32)), output)
params = Dict(x=>Param{Tensor}(y, gradients(mapt(x->x.x, output), params = Dict(x=>Param{Tensor}(y, gradients(mapt(x->x.x, output),
y, mapt(x->x.Δx, output))) for (x, y) in params) y, mapt(x->x.Δx, output)))
for (x, y) in params)
inputs = mapt(x->Param{Tensor}(x, gradients(mapt(x->x.x, output), inputs = mapt(x->Param{Tensor}(x, gradients(mapt(x->x.x, output),
x, mapt(x->x.Δx, output))), inputs) x, mapt(x->x.Δx, output))),
inputs)
run(session, global_variables_initializer()) run(session, global_variables_initializer())
Exec(session, inputs, output, params, stacks) Exec(session, inputs, output, params, stacks)
end end