Skip to content

GitLab

"macapp/webpack.plugins.ts" did not exist on "32c6d8e6cc38b9f51361fa646152f21a7da3ebc2"
Commit 10806935 authored by Liangzhe Yuan's avatar Liangzhe Yuan Committed by A. Unique TensorFlower
Browse files

Fix the audio dimensions for eval input. 

PiperOrigin-RevId: 420825486
parent cf1f8a2a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/dataloaders/video_input.py
View file @ 10806935
...@@ -361,7 +361,7 @@ class Parser(parser.Parser): ...@@ -361,7 +361,7 @@ class Parser(parser.Parser):
audio = decoded_tensors[self._audio_feature] audio = decoded_tensors[self._audio_feature]
audio = tf.cast(audio, dtype=self._dtype) audio = tf.cast(audio, dtype=self._dtype)
audio = preprocess_ops_3d.sample_sequence( 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) audio = tf.ensure_shape(audio, self._audio_shape)
features['audio'] = audio features['audio'] = audio
......
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