Internal change

PiperOrigin-RevId: 380687139

official/vision/beta/modeling/backbones/spinenet.py

@@ -393,8 +393,10 @@ class SpineNet(tf.keras.Model):
               block_spec.level))
         if (block_spec.level < self._min_level or
             block_spec.level > self._max_level):
-          raise ValueError('Output level is out of range [{}, {}]'.format(
-              self._min_level, self._max_level))
+          logging.warning(
+              'SpineNet output level out of range [min_level, max_level] = '
+              '[%s, %s] will not be used for further processing.',
+              self._min_level, self._max_level)
         endpoints[str(block_spec.level)] = x
 
     return endpoints

official/vision/beta/modeling/backbones/spinenet_test.py

@@ -24,17 +24,16 @@ from official.vision.beta.modeling.backbones import spinenet
 class SpineNetTest(parameterized.TestCase, tf.test.TestCase):
 
   @parameterized.parameters(
-      (128, 0.65, 1, 0.5, 128),
-      (256, 1.0, 1, 0.5, 256),
-      (384, 1.0, 2, 0.5, 256),
-      (512, 1.0, 3, 1.0, 256),
-      (640, 1.3, 4, 1.0, 384),
+      (128, 0.65, 1, 0.5, 128, 4, 6),
+      (256, 1.0, 1, 0.5, 256, 3, 6),
+      (384, 1.0, 2, 0.5, 256, 4, 7),
+      (512, 1.0, 3, 1.0, 256, 3, 7),
+      (640, 1.3, 4, 1.0, 384, 3, 7),
   )
   def test_network_creation(self, input_size, filter_size_scale, block_repeats,
-                            resample_alpha, endpoints_num_filters):
+                            resample_alpha, endpoints_num_filters, min_level,
+                            max_level):
     """Test creation of SpineNet models."""
-    min_level = 3
-    max_level = 7
 
     tf.keras.backend.set_image_data_format('channels_last')