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