Skip to content
This repository was archived by the owner on Nov 3, 2022. It is now read-only.

Commit cb45588

Browse files
committed
Fix the default shape of NASNet
1 parent 69b58ff commit cb45588

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras_applications/nasnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ def NASNet(input_shape=None,
161161
default_size=default_size,
162162
min_size=32,
163163
data_format=backend.image_data_format(),
164-
require_flatten=False,
164+
require_flatten=include_top,
165165
weights=weights)
166166

167167
if backend.image_data_format() != 'channels_last':

0 commit comments

Comments
 (0)