Skip to content

GitLab

Commit 8b35f33e authored by Ron Shapiro's avatar Ron Shapiro Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 475113897
parent 8955b748
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
official/vision/modeling/factory.py
View file @ 8b35f33e
......@@ -260,7 +260,7 @@ def build_retinanet(
model_config: retinanet_cfg.RetinaNet,
l2_regularizer: Optional[tf.keras.regularizers.Regularizer] = None,
backbone: Optional[tf.keras.Model] = None,
decoder: Optional[tf.keras.regularizers.Regularizer] = None
decoder: Optional[tf.keras.Model] = None
) -> tf.keras.Model:
"""Builds RetinaNet model."""
norm_activation_config = model_config.norm_activation
......@@ -335,8 +335,8 @@ def build_segmentation_model(
input_specs: tf.keras.layers.InputSpec,
model_config: segmentation_cfg.SemanticSegmentationModel,
l2_regularizer: Optional[tf.keras.regularizers.Regularizer] = None,
backbone: Optional[tf.keras.regularizers.Regularizer] = None,
decoder: Optional[tf.keras.regularizers.Regularizer] = None
backbone: Optional[tf.keras.Model] = None,
decoder: Optional[tf.keras.Model] = None
) -> tf.keras.Model:
"""Builds Segmentation model."""
norm_activation_config = model_config.norm_activation
......
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