Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "feea48cd06e5054cdc3778e03f0b26634b9ed718"
Commit 5ab89ea9 authored by Ruoxin Sang's avatar Ruoxin Sang Committed by A. Unique TensorFlower
Browse files

Specify the axis explicitly when using `tf.squeueze` to avoid unknown result... 

Specify the axis explicitly when using `tf.squeueze` to avoid unknown result shape when the input contains dynamic shape dimensions.

PiperOrigin-RevId: 457858716
parent e6248341
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/projects/yt8m/dataloaders/yt8m_input.py
View file @ 5ab89ea9
...@@ -424,8 +424,8 @@ class PostBatchProcessor(): ...@@ -424,8 +424,8 @@ class PostBatchProcessor():
[-1, self.num_classes]) [-1, self.num_classes])
else: else:
video_matrix = tf.squeeze(video_matrix) video_matrix = tf.squeeze(video_matrix, axis=1)
labels = tf.squeeze(labels) labels = tf.squeeze(labels, axis=1)
batched_tensors = { batched_tensors = {
"video_matrix": video_matrix, "video_matrix": video_matrix,
......
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