From 40996874de5ae32ee2f2683a8b4010a5a3616130 Mon Sep 17 00:00:00 2001 From: whyguu Date: Mon, 13 Nov 2017 13:34:37 +0800 Subject: [PATCH] Update googLeNet.py --- googLeNet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/googLeNet.py b/googLeNet.py index 5c57207..c7901a1 100644 --- a/googLeNet.py +++ b/googLeNet.py @@ -45,7 +45,7 @@ def define_model(weight_path = None): drop1 = Dropout(rate=0.4)(averagepool1_7x7_s1) - linear = Dense(units=1000, activation='softmax', kernel_regularizer=l2(0.01))(keras.layers.core.Flatten(drop1)) + linear = Dense(units=1000, activation='softmax', kernel_regularizer=l2(0.01))(keras.layers.core.Flatten()(drop1)) last = linear