Dimensions of training data
JSHZT opened this issue · 1 comments
JSHZT commented
What is the shape of the data for each iteration of your dataloader iterator? Is it (BxTxVx3)?
alexanderrichard commented
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.