diff --git a/resnext.py b/resnext.py index 4f0ed23..ad54834 100644 --- a/resnext.py +++ b/resnext.py @@ -101,7 +101,7 @@ def ResNext(input_shape=None, depth=29, cardinality=8, width=64, weight_decay=5e default_size=32, min_size=8, 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)