Enable input spec checking for Functional models.

PiperOrigin-RevId: 324625967

Enable input spec checking for Functional models.
PiperOrigin-RevId: 324625967
db87dd5b · Francois Chollet · TF Object Detection Team · d5776810 · db87dd5b
Commit db87dd5b authored Aug 03, 2020 by Francois Chollet Committed by TF Object Detection Team Aug 03, 2020
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

research/object_detection/models/keras_models/resnet_v1_tf2_test.py ...bject_detection/models/keras_models/resnet_v1_tf2_test.py +2 -2

No files found.
--- a/research/object_detection/models/keras_models/resnet_v1_tf2_test.py
+++ b/research/object_detection/models/keras_models/resnet_v1_tf2_test.py
@@ -202,9 +202,9 @@ class ResnetShapeTest(test_case.TestCase, parameterized.TestCase):
      })
  def test_output_shapes(self, resnet_type, output_layer_names):
    if resnet_type == 'resnet_v1_34':
-      model = resnet_v1.resnet_v1_34(weights=None)
+      model = resnet_v1.resnet_v1_34(input_shape=(64, 64, 3), weights=None)
    else:
-      model = resnet_v1.resnet_v1_18(weights=None)
+      model = resnet_v1.resnet_v1_18(input_shape=(64, 64, 3), weights=None)
    outputs = [
        model.get_layer(output_layer_name).output
        for output_layer_name in output_layer_names