Skip to content

GitLab

Commit 7e18563f authored by Xianzhi Du's avatar Xianzhi Du Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 372184016
parent 52d9dfa1
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
official/vision/beta/modeling/backbones/spinenet_mobile.py
View file @ 7e18563f
...@@ -320,6 +320,9 @@ class SpineNetMobile(tf.keras.Model): ...@@ -320,6 +320,9 @@ class SpineNetMobile(tf.keras.Model):
endpoints = {} endpoints = {}
for i, block_spec in enumerate(self._block_specs): for i, block_spec in enumerate(self._block_specs):
# Update block level if it is larger than max_level to avoid building
# blocks smaller than requested.
block_spec.level = min(block_spec.level, self._max_level)
# Find out specs for the target block. # Find out specs for the target block.
target_width = int(math.ceil(input_width / 2**block_spec.level)) target_width = int(math.ceil(input_width / 2**block_spec.level))
target_num_filters = int(FILTER_SIZE_MAP[block_spec.level] * target_num_filters = int(FILTER_SIZE_MAP[block_spec.level] *
...@@ -392,8 +395,9 @@ class SpineNetMobile(tf.keras.Model): ...@@ -392,8 +395,9 @@ class SpineNetMobile(tf.keras.Model):
block_spec.level)) block_spec.level))
if (block_spec.level < self._min_level or if (block_spec.level < self._min_level or
block_spec.level > self._max_level): block_spec.level > self._max_level):
raise ValueError('Output level is out of range [{}, {}]'.format( logging.warning(
self._min_level, self._max_level)) 'SpineNet output level out of range [min_level, max_levle] = [%s, %s] will not be used for further processing.',
self._min_level, self._max_level)
endpoints[str(block_spec.level)] = x endpoints[str(block_spec.level)] = x
return endpoints return endpoints
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment