From 6e829a7b44496bd08814d461858af9f86afc8d52 Mon Sep 17 00:00:00 2001 From: Anon George Date: Wed, 17 Jan 2018 16:29:06 +0530 Subject: [PATCH] include_top changed to require_flatten include_top=include_top does not work in new keras version and had to be changed to require_flatten= include_top --- vgg16.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vgg16.py b/vgg16.py index 373c5c2..b2a959a 100644 --- a/vgg16.py +++ b/vgg16.py @@ -99,7 +99,7 @@ def VGG16(include_top=True, weights='imagenet', default_size=224, min_size=48, 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)