When I use transforms.FiveCrop(20) in the custom dataset, Then pytorch dataloader gives me (128,5,3,20,20) shapes of batches where 128 is the batchsize, 5 comes from croping an image at 5 different places and remaining is channel, width and height.
What I want is that I get only 128 images in the batch.