Skip to content

GitLab

Commit 62767d6e authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 374926672
parent 0af0abe9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/vision/beta/modeling/factory.py
View file @ 62767d6e
......@@ -76,6 +76,7 @@ def build_maskrcnn(
backbone_config=model_config.backbone,
norm_activation_config=norm_activation_config,
l2_regularizer=l2_regularizer)
backbone(tf.keras.Input(input_specs.shape[1:]))
decoder = decoder_factory.build_decoder(
input_specs=backbone.output_specs,
......@@ -250,6 +251,7 @@ def build_retinanet(
backbone_config=model_config.backbone,
norm_activation_config=norm_activation_config,
l2_regularizer=l2_regularizer)
backbone(tf.keras.Input(input_specs.shape[1:]))
decoder = decoder_factory.build_decoder(
input_specs=backbone.output_specs,
......
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