Internal change

PiperOrigin-RevId: 406237483

Internal change
PiperOrigin-RevId: 406237483
067c893e · A. Unique TensorFlower · fed7723c · 067c893e
Commit 067c893e authored Oct 28, 2021 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

official/vision/beta/modeling/retinanet_model.py official/vision/beta/modeling/retinanet_model.py +3 -3

No files found.
--- a/official/vision/beta/modeling/retinanet_model.py
+++ b/official/vision/beta/modeling/retinanet_model.py
@@ -123,9 +123,9 @@ class RetinaNetModel(tf.keras.Model):
          {'backbone_{}'.format(k): v for k, v in features.items()})
    if self.decoder:
      features = self.decoder(features)
-      if output_intermediate_features:
-        outputs.update(
-            {'decoder_{}'.format(k): v for k, v in features.items()})
+    if output_intermediate_features:
+      outputs.update(
+          {'decoder_{}'.format(k): v for k, v in features.items()})

    # Dense prediction. `raw_attributes` can be empty.
    raw_scores, raw_boxes, raw_attributes = self.head(features)