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

fix tensorflow dep

Browse Source
This commit is contained in:
Mike J Innes 2017-04-27 12:48:11 +01:00
parent 52a7199d10
commit f4acd286c8
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/integration-tf.jl
View File

@ -38,7 +38,7 @@ train_step = train.minimize(train.AdamOptimizer(1e-4), cross_entropy)
accuracy = reduce_mean(cast(indmax(y, 2) .== indmax(y′, 2), Float32)) accuracy = reduce_mean(cast(indmax(y, 2) .== indmax(y′, 2), Float32))
run(session, initialize_all_variables()) run(session, global_variables_initializer())
@progress for i in 1:1000 @progress for i in 1:1000
batch = next_batch(loader, 50) batch = next_batch(loader, 50)

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

@ -9,7 +9,7 @@ end
function makesession(model, inputs; session = Session(Graph())) function makesession(model, inputs; session = Session(Graph()))
params, stacks, output = tograph(model, inputs...) params, stacks, output = tograph(model, inputs...)
run(session, initialize_all_variables()) run(session, global_variables_initializer())
Model(model, session, params, stacks, inputs, output) Model(model, session, params, stacks, inputs, output)
end end

2
test/backend/tensorflow.jl
View File

@ -13,7 +13,7 @@ dt = tf(d)
sess = TensorFlow.Session() sess = TensorFlow.Session()
X = placeholder(Float32) X = placeholder(Float32)
Y = Tensor(d, X) Y = Tensor(d, X)
run(sess, initialize_all_variables()) run(sess, global_variables_initializer())
@test run(sess, Y, Dict(X=>Float32.(xs))) ≈ d(xs) @test run(sess, Y, Dict(X=>Float32.(xs))) ≈ d(xs)
end end