Commit d1baa18c authored by Mostafa Rahmani's avatar Mostafa Rahmani Committed by GitHub
Browse files

Update cifar10_input.py

bug fix for contrib.deprecated eliminatation in tf version 12.
parent 62cea658
...@@ -132,7 +132,7 @@ def _generate_image_and_label_batch(image, label, min_queue_examples, ...@@ -132,7 +132,7 @@ def _generate_image_and_label_batch(image, label, min_queue_examples,
capacity=min_queue_examples + 3 * batch_size) capacity=min_queue_examples + 3 * batch_size)
# Display the training images in the visualizer. # Display the training images in the visualizer.
tf.contrib.deprecated.image_summary('images', images) tf.image_summary('images', images)
return images, tf.reshape(label_batch, [batch_size]) return images, tf.reshape(label_batch, [batch_size])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment