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

style improvement

Browse Source
This commit is contained in:
Mike J Innes 2016-12-20 17:33:14 +00:00
parent 1b5b28897c
commit b580c2e4a7
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/backend/tensorflow/recurrent.jl
View File

@ -11,16 +11,17 @@ function makesession(model::Flux.Unrolled)
sess = Session(Graph()) sess = Session(Graph())
input = placeholder(Float32) input = placeholder(Float32)
inputs = TensorFlow.unpack(input, num = model.steps, axis = 1) inputs = TensorFlow.unpack(input, num = model.steps, axis = 1)
params, stacks, outputs, instates, outstates = [], [], [], [], [] let params, stacks, outputs, instates, outstates
if model.stateful if model.stateful
instates = [placeholder(Float32) for _ in model.state] instates = [placeholder(Float32) for _ in model.state]
params, stacks, (outstates, outputs) = tograph(model, cgroup(instates...), cgroup(inputs...)) params, stacks, (outstates, outputs) = tograph(model, cgroup(instates...), cgroup(inputs...))
else else
params, stacks, outputs = tograph(model, cgroup(inputs...)) params, stacks, outputs = tograph(model, cgroup(inputs...))
end
output = TensorFlow.pack(outputs, axis = 1)
run(sess, initialize_all_variables())
sess, params, stacks, (instates, input), (outstates, output)
end end
output = TensorFlow.pack(outputs, axis = 1)
run(sess, initialize_all_variables())
sess, params, stacks, (instates, input), (outstates, output)
end end
function tf(model::Flux.Unrolled) function tf(model::Flux.Unrolled)