Internal change

PiperOrigin-RevId: 368070382

Internal change
PiperOrigin-RevId: 368070382
0338ca4b · Tianjian Meng · A. Unique TensorFlower · 1bbd359d · 0338ca4b
Commit 0338ca4b authored Apr 12, 2021 by Tianjian Meng Committed by A. Unique TensorFlower Apr 12, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/modeling/backbones/efficientnet.py official/vision/beta/modeling/backbones/efficientnet.py +1 -1

No files found.
--- a/official/vision/beta/modeling/backbones/efficientnet.py
+++ b/official/vision/beta/modeling/backbones/efficientnet.py
@@ -184,7 +184,7 @@ class EfficientNet(tf.keras.Model):
        endpoint_level += 1

    # Build output specs for downstream tasks.
-    self._output_specs = {l: endpoints[l].get_shape for l in endpoints.keys()}
+    self._output_specs = {l: endpoints[l].get_shape() for l in endpoints}

    # Build the final conv for classification.
    x = layers.Conv2D(