Update code with require_flatten

GKalliatakis · Jan 3, 2018 · 50648dd · 50648dd
1 parent b64cc00
commit 50648dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/resnet101.py b/resnet101.py
@@ -285,7 +285,7 @@ def ResNet101(include_top=True, weights='imagenet',
                                       default_size=img_size,
                                       min_size=197,
                                       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)