Internal change

PiperOrigin-RevId: 464827311

official/projects/yt8m/modeling/yt8m_model.py

@@ -85,8 +85,9 @@ class DbofModel(tf.keras.Model):
     feature_size = input_specs.shape[-1]
     # shape 'excluding' batch_size
     model_input = tf.keras.Input(shape=self._input_specs.shape[1:])
-    tf.summary.histogram("input_hist", model_input)
-    input_data = model_input
+    # normalize input features
+    input_data = tf.nn.l2_normalize(model_input, -1)
+    tf.summary.histogram("input_hist", input_data)
 
     # configure model
     if params.add_batch_norm:

official/projects/yt8m/tasks/yt8m_task.py

@@ -168,10 +168,6 @@ class YT8MTask(base_task.Task):
     features, labels = inputs['video_matrix'], inputs['labels']
     num_frames = inputs['num_frames']
 
-    # Normalize input features.
-    feature_dim = len(features.shape) - 1
-    features = tf.nn.l2_normalize(features, feature_dim)
-
     # sample random frames / random sequence
     num_frames = tf.cast(num_frames, tf.float32)
     sample_frames = self.task_config.train_data.num_frames
@@ -244,10 +240,6 @@ class YT8MTask(base_task.Task):
     features, labels = inputs['video_matrix'], inputs['labels']
     num_frames = inputs['num_frames']
 
-    # Normalize input features.
-    feature_dim = len(features.shape) - 1
-    features = tf.nn.l2_normalize(features, feature_dim)
-
     # sample random frames (None, 5, 1152) -> (None, 30, 1152)
     sample_frames = self.task_config.validation_data.num_frames
     if self.task_config.model.sample_random_frames: