From f13895b9925575379236e181cbfceec73d59b7c6 Mon Sep 17 00:00:00 2001
From: Xianzhi Du <xianzhi@google.com>
Date: Mon, 21 Jun 2021 16:16:32 -0700
Subject: [PATCH] Internal change

PiperOrigin-RevId: 380687139
---
 .../vision/beta/modeling/backbones/spinenet.py    |  6 ++++--
 .../beta/modeling/backbones/spinenet_test.py      | 15 +++++++--------
 2 files changed, 11 insertions(+), 10 deletions(-)

diff --git a/official/vision/beta/modeling/backbones/spinenet.py b/official/vision/beta/modeling/backbones/spinenet.py
index 90df460a4..db04ed4f2 100644
--- a/official/vision/beta/modeling/backbones/spinenet.py
+++ b/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
diff --git a/official/vision/beta/modeling/backbones/spinenet_test.py b/official/vision/beta/modeling/backbones/spinenet_test.py
index 51061030f..e3a036a6e 100644
--- a/official/vision/beta/modeling/backbones/spinenet_test.py
+++ b/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')
 
-- 
GitLab