I have created a custom Dataset
and when passing it to a DataLoader
during training, I am getting
RuntimeError: expected stride to be a single integer value or a list . . .
I have tried to unsqueeze
the inputs with no success. This issue does not persist when I instead iterate over the Dataset
itself during training (which is obviously inefficient as I am unable to take advantage of batches and workers). The model is resnet50
.
What are some steps I can take to overcome this issue?