"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "98c9aac1d5476e58a10eff37111af4b06fd49add"
Commit 4d38b150 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 406250903
parent 37103aec
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
from typing import Any, Mapping, Optional from typing import Any, Mapping, Optional
# Import libraries # Import libraries
from absl import logging
import tensorflow as tf import tensorflow as tf
from official.modeling import hyperparams from official.modeling import hyperparams
...@@ -108,6 +108,7 @@ class FPN(tf.keras.Model): ...@@ -108,6 +108,7 @@ class FPN(tf.keras.Model):
bn_axis = 1 bn_axis = 1
# Get input feature pyramid from backbone. # Get input feature pyramid from backbone.
logging.info('FPN input_specs: %s', input_specs)
inputs = self._build_input_pyramid(input_specs, min_level) inputs = self._build_input_pyramid(input_specs, min_level)
backbone_max_level = min(int(max(inputs.keys())), max_level) backbone_max_level = min(int(max(inputs.keys())), max_level)
......
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