Skip to content

GitLab

Commit 2f689ffc authored by Francois Chollet's avatar Francois Chollet Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 324625967
parent eb47d489
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
official/vision/detection/modeling/retinanet_model.py
View file @ 2f689ffc
...@@ -59,11 +59,8 @@ class RetinanetModel(base_model.Model): ...@@ -59,11 +59,8 @@ class RetinanetModel(base_model.Model):
self._transpose_input = params.train.transpose_input self._transpose_input = params.train.transpose_input
assert not self._transpose_input, 'Transpose input is not supported.' assert not self._transpose_input, 'Transpose input is not supported.'
# Input layer. # Input layer.
input_shape = (
params.retinanet_parser.output_size +
[params.retinanet_parser.num_channels])
self._input_layer = tf.keras.layers.Input( self._input_layer = tf.keras.layers.Input(
shape=input_shape, name='', shape=(None, None, params.retinanet_parser.num_channels), name='',
dtype=tf.bfloat16 if self._use_bfloat16 else tf.float32) dtype=tf.bfloat16 if self._use_bfloat16 else tf.float32)
def build_outputs(self, inputs, mode): def build_outputs(self, inputs, mode):
......
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