standard input_shape if include_top is false (#18954)

keras-team · Dec 17, 2023 · 565953a · 565953a
1 parent d550552
commit 565953a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras/applications/nasnet.py b/keras/applications/nasnet.py
@@ -384,6 +384,8 @@ def NASNetMobile(
             "at this time due to an outstanding bug. "
             "If interested, please open a PR."
         )
+    if (include_top == False) and (input_shape == None):
+        input_shape = (224, 224, 3)
     return NASNet(
         input_shape,
         penultimate_filters=1056,