DataLoader IndexError: Dimension out of range during training

[Pinole]

``Hello,

I am currently encountering an issue during the training process of my model. The error message that I receive is as follows:

IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

This error occurs when enumerating over the DataLoader in my training loop:

for batch_idx, (x, x_lengths, spec, spec_lengths, y, y_lengths, speakers, emo) in enumerate(train_loader):

The batch data seems to be of the correct shape when I print it out just before the loop:

batch[0][0].shape: torch.Size([17])
batch[0][1].shape: torch.Size([513])
batch[0][2].shape: torch.Size([1, 66150])
batch[0][3].shape: torch.Size([1])
batch[0][4].shape: torch.Size([1024])

The problem seems to occur when the collate_fn function of the DataLoader tries to create a LongTensor from the sizes of the batch data:

torch.LongTensor([x[1].size(1) for x in batch])

I have been trying to debug this issue, but I am currently stuck. Any help or pointers would be greatly appreciated.

Regard

[jamesD33064]

Same question QQ