Skip to content

GitLab

Commit b792005c authored by Alan Chiao's avatar Alan Chiao Committed by A. Unique TensorFlower
Browse files

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

PiperOrigin-RevId: 330857499
parent 38db214b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/benchmark/models/resnet_imagenet_main.py
View file @ b792005c
...@@ -145,8 +145,7 @@ def run(flags_obj): ...@@ -145,8 +145,7 @@ def run(flags_obj):
# output format should be same as the keras backend image data format or just # output format should be same as the keras backend image data format or just
# channel-last format. # channel-last format.
use_keras_image_data_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( train_input_dataset = input_fn(
is_training=True, is_training=True,
......
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