Skip to content

GitLab

Commit 0338ca4b authored by Tianjian Meng's avatar Tianjian Meng Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 368070382
parent 1bbd359d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/modeling/backbones/efficientnet.py
View file @ 0338ca4b
...@@ -184,7 +184,7 @@ class EfficientNet(tf.keras.Model): ...@@ -184,7 +184,7 @@ class EfficientNet(tf.keras.Model):
endpoint_level += 1 endpoint_level += 1
# Build output specs for downstream tasks. # 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. # Build the final conv for classification.
x = layers.Conv2D( x = layers.Conv2D(
......
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