fix: user input for data_format is ignored (tensorflow#3871)

domschl · karmel · commit 3a6fff33f8be · 2018-04-04T09:23:17.000-07:00
diff --git a/official/mnist/mnist_eager.py b/official/mnist/mnist_eager.py
@@ -107,7 +107,7 @@ def main(argv):
     (device, data_format) = ('/cpu:0', 'channels_last')
   # If data_format is defined in FLAGS, overwrite automatically set value.
   if flags.data_format is not None:
-    data_format = data_format
+    data_format = flags.data_format
   print('Using device %s, and data format %s.' % (device, data_format))
 
   # Load the datasets
