Merge pull request tensorflow#3149 from MikalaiDrabovich/patch-1

Fix tensorflow#3148
donyang · Jan 13, 2018 · 1887a5f · 1887a5f
2 parents aafe60d + 6487fa7
commit 1887a5f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/official/mnist/dataset.py b/official/mnist/dataset.py
@@ -33,7 +33,7 @@ def read32(bytestream):
 
 def check_image_file_header(filename):
   """Validate that filename corresponds to images for the MNIST dataset."""
-  with tf.gfile.Open(filename) as f:
+  with tf.gfile.Open(filename,'rb') as f:
     magic = read32(f)
     num_images = read32(f)
     rows = read32(f)
@@ -49,7 +49,7 @@ def check_image_file_header(filename):
 
 def check_labels_file_header(filename):
   """Validate that filename corresponds to labels for the MNIST dataset."""
-  with tf.gfile.Open(filename) as f:
+  with tf.gfile.Open(filename,'rb') as f:
     magic = read32(f)
     num_items = read32(f)
     if magic != 2049: