From 50648dd7e1bc927a8694736f455502f813e24240 Mon Sep 17 00:00:00 2001 From: Grigorios Kalliatakis Date: Wed, 3 Jan 2018 12:52:12 +0000 Subject: [PATCH] Update code with require_flatten --- resnet101.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/resnet101.py b/resnet101.py index 3848f0d..8ad6a91 100644 --- 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)