Closed JSHZT closed 2 years ago
What is the shape of the data for each iteration of your dataloader iterator? Is it (BxTxVx3)?
The datareader returns elements of shape T x V x 3. So, if you wrap this into a torch DataLoader, you will get batches of shape B x T x V x 3.
What is the shape of the data for each iteration of your dataloader iterator? Is it (BxTxVx3)?