ESRGAN how to use self-dataset?

[Edenkut]

Hi, I am a newbie and I am trying to train using the MRI2D dataset. My data set has saved 2D slices in a matrix form with the shape [500, 3, 64, 64]. When using the dataloader, I got an error: Expected 4-dimensional input for 4-dimensional weight 64 3 3 3 but got 5-dimensional input of size [4, 500, 3, 64, 64] instead. What should I do? 500 is the number of 2Dmri pictures, and 3 is the number of channels. I found that it reads the 4D dataset and batches it, adding a latitude. That’s why there are 4 more. What should I do to make a batch in my 500 slices. [4,3,64,64] is I expect, but got [4,500,3,64,64]