diff --git a/inception_v3.py b/inception_v3.py index dbb34d1..33728f5 100644 --- a/inception_v3.py +++ b/inception_v3.py @@ -154,7 +154,7 @@ def InceptionV3(include_top=True, default_size=299, min_size=139, data_format=K.image_data_format(), - include_top=include_top) + require_flatten=include_top) if input_tensor is None: img_input = Input(shape=input_shape)