Can you please add the code that’s causing the error.
It seems like the error is caused by the default collate function, which is responsible for taking the lists that you input and batching them into torch tensors. I would presume that ‘jfif’ is a key in your DataSet class which is missing in some of the objects, which the DataLoader is picking up on.
So the code that takes the dataset class and constructs a DataLoader out of it will be very helpful.