Deprecated tf.session removed in mnist_tflite.py

2019-07-10 14:42:30 +05:30 · 2019-07-10 14:42:30 +05:30 · 6de3421b2e
commit 6de3421b2e
parent bc73b9cde7
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/lite/tutorials/mnist_tflite.py
+++ b/tensorflow/lite/tutorials/mnist_tflite.py
@ -33,7 +33,7 @@ flags = flags.FLAGS

 def test_image_generator():
  # Generates an iterator over images
-  with tf.Session() as sess:
+  with tf.compat.v1.Session() as sess:
    input_data = tf.compat.v1.data.make_one_shot_iterator(dataset.test(
        flags.data_dir)).get_next()
    try: