From f4acd286c86eca5af0e472fa3af7a84cfadfef79 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Thu, 27 Apr 2017 12:48:11 +0100 Subject: [PATCH] fix tensorflow dep --- examples/integration-tf.jl | 2 +- src/backend/tensorflow/model.jl | 2 +- test/backend/tensorflow.jl | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/integration-tf.jl b/examples/integration-tf.jl index 12faa4d9..eddfb834 100644 --- a/examples/integration-tf.jl +++ b/examples/integration-tf.jl @@ -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)) -run(session, initialize_all_variables()) +run(session, global_variables_initializer()) @progress for i in 1:1000 batch = next_batch(loader, 50) diff --git a/src/backend/tensorflow/model.jl b/src/backend/tensorflow/model.jl index ed869504..2ec5ebe9 100644 --- a/src/backend/tensorflow/model.jl +++ b/src/backend/tensorflow/model.jl @@ -9,7 +9,7 @@ end function makesession(model, inputs; session = Session(Graph())) params, stacks, output = tograph(model, inputs...) - run(session, initialize_all_variables()) + run(session, global_variables_initializer()) Model(model, session, params, stacks, inputs, output) end diff --git a/test/backend/tensorflow.jl b/test/backend/tensorflow.jl index 459402cd..81a9ad98 100644 --- a/test/backend/tensorflow.jl +++ b/test/backend/tensorflow.jl @@ -13,7 +13,7 @@ dt = tf(d) sess = TensorFlow.Session() X = placeholder(Float32) Y = Tensor(d, X) - run(sess, initialize_all_variables()) + run(sess, global_variables_initializer()) @test run(sess, Y, Dict(X=>Float32.(xs))) ≈ d(xs) end