Fix syntax issue causing input shape to not be set properly for resnet.

PiperOrigin-RevId: 330857499

Fix syntax issue causing input shape to not be set properly for resnet.
PiperOrigin-RevId: 330857499
b792005c · Alan Chiao · A. Unique TensorFlower · 38db214b · b792005c
Commit b792005c authored Sep 09, 2020 by Alan Chiao Committed by A. Unique TensorFlower Sep 09, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

official/benchmark/models/resnet_imagenet_main.py official/benchmark/models/resnet_imagenet_main.py +1 -2

No files found.
--- a/official/benchmark/models/resnet_imagenet_main.py
+++ b/official/benchmark/models/resnet_imagenet_main.py
@@ -145,8 +145,7 @@ def run(flags_obj):
  # output format should be same as the keras backend image data format or just
  # channel-last format.
  use_keras_image_data_format = \
-    (flags_obj.model == 'mobilenet' or
+    (flags_obj.model == 'mobilenet' or flags_obj.model == 'mobilenet_pretrained')
-     flags_obj.model == 'mobilenet_pretrained')
  train_input_dataset = input_fn(
      is_training=True,