Skip to content

GitLab

Commit 3a6fff33 authored by Dominik Schlösser's avatar Dominik Schlösser Committed by Karmel Allison
Browse files

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

parent 2041d5ca
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/mnist/mnist_eager.py
View file @ 3a6fff33
...@@ -107,7 +107,7 @@ def main(argv): ...@@ -107,7 +107,7 @@ def main(argv):
(device, data_format) = ('/cpu:0', 'channels_last') (device, data_format) = ('/cpu:0', 'channels_last')
# If data_format is defined in FLAGS, overwrite automatically set value. # If data_format is defined in FLAGS, overwrite automatically set value.
if flags.data_format is not None: 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)) print('Using device %s, and data format %s.' % (device, data_format))
# Load the datasets # Load the datasets
......
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