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

Commit 7a03145

Browse files
authored
Set the default shape for NASNet when include_top=False (#62)
* Set the default shape for NASNet when `include_top=False` * Disable NASNet tests for `include_top=False`
1 parent 43ac53e commit 7a03145

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

keras_applications/nasnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def NASNet(input_shape=None,
164164
default_size=default_size,
165165
min_size=32,
166166
data_format=backend.image_data_format(),
167-
require_flatten=include_top,
167+
require_flatten=True,
168168
weights=weights)
169169

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

tests/applications_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -269,8 +269,8 @@ def test_nasnet():
269269
app, last_dim = NASNET_LIST[0] # NASNetLarge is too heavy to test on Travis
270270
module = nasnet
271271
_test_application_basic(app, module=module)
272-
_test_application_notop(app, last_dim)
273-
_test_application_variable_input_channels(app, last_dim)
272+
# _test_application_notop(app, last_dim)
273+
# _test_application_variable_input_channels(app, last_dim)
274274
_test_app_pooling(app, last_dim)
275275

276276

0 commit comments

Comments
 (0)