Fix the audio dimensions for eval input.

PiperOrigin-RevId: 420825486
brijs · Jan 10, 2022 · 1080693 · 1080693
1 parent cf1f8a2
commit 1080693
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/official/vision/beta/dataloaders/video_input.py b/official/vision/beta/dataloaders/video_input.py
@@ -361,7 +361,7 @@ def _parse_eval_data(
       audio = decoded_tensors[self._audio_feature]
       audio = tf.cast(audio, dtype=self._dtype)
       audio = preprocess_ops_3d.sample_sequence(
-          audio, 20, random=False, stride=1)
+          audio, self._audio_shape[0], random=False, stride=1)
       audio = tf.ensure_shape(audio, self._audio_shape)
       features['audio'] = audio