Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 368040234
parent 6f917c69
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/modeling/classification_model.py
View file @ ecbc3fd8
...@@ -65,7 +65,7 @@ class ClassificationModel(tf.keras.Model): ...@@ -65,7 +65,7 @@ class ClassificationModel(tf.keras.Model):
norm = tf.keras.layers.BatchNormalization norm = tf.keras.layers.BatchNormalization
axis = -1 if tf.keras.backend.image_data_format() == 'channels_last' else 1 axis = -1 if tf.keras.backend.image_data_format() == 'channels_last' else 1
inputs = tf.keras.Input(shape=input_specs.shape[1:]) inputs = tf.keras.Input(shape=input_specs.shape[1:], name=input_specs.name)
endpoints = backbone(inputs) endpoints = backbone(inputs)
x = endpoints[max(endpoints.keys())] x = endpoints[max(endpoints.keys())]
......
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