Internal change

PiperOrigin-RevId: 422928888

official/vision/beta/projects/centernet/configs/experiments/coco-centernet-hourglass-tpu.yaml

@@ -59,11 +59,11 @@ task:
     input_path: 'coco/val*'
     drop_remainder: false
     dtype: 'bfloat16'
-    global_batch_size: 16
+    global_batch_size: 64
     is_training: false
 trainer:
   train_steps: 140000
-  validation_steps: 78  # 5000 / 16
+  validation_steps: 78  # 5000 / 64
   steps_per_loop: 924  # 118287 / 128
   validation_interval: 924
   summary_interval: 924

official/vision/beta/projects/centernet/modeling/heads/centernet_head.py

@@ -61,7 +61,6 @@ class CenterNetHead(tf.keras.Model):
     self._heatmap_bias = heatmap_bias
     self._num_inputs = len(input_levels)
 
-    input_levels = sorted(self._input_specs.keys())
     inputs = {level: tf.keras.layers.Input(shape=self._input_specs[level][1:])
               for level in input_levels}
     outputs = {}