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
......@@ -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
......
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