IndexError: index 10 is out of bounds for dimension 0 with size 10

[martinoywa]

I’ve been stuck here for a while now. Any help.

[martinoywa]

code: batch_size = 30

[ptrblck]

The last batch given by your DataLoader might be smaller, if the length of your Dataset is not divisible by the batch size without a remainder.
Try to get the current batch size inside the training loop:

for data, target in testloader:
    batch_size = data.size(0)
    ...
    for i in range(batch_size):
        ...

[martinoywa]

I worked perfectly!!! Thank you!