Skip to content

Commit 1080693

Browse files
yuanliangzhetensorflower-gardener
authored andcommitted
Fix the audio dimensions for eval input.
PiperOrigin-RevId: 420825486
1 parent cf1f8a2 commit 1080693

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

official/vision/beta/dataloaders/video_input.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -361,7 +361,7 @@ def _parse_eval_data(
361361
audio = decoded_tensors[self._audio_feature]
362362
audio = tf.cast(audio, dtype=self._dtype)
363363
audio = preprocess_ops_3d.sample_sequence(
364-
audio, 20, random=False, stride=1)
364+
audio, self._audio_shape[0], random=False, stride=1)
365365
audio = tf.ensure_shape(audio, self._audio_shape)
366366
features['audio'] = audio
367367

0 commit comments

Comments
 (0)