I actually don’t use scipy, I only use skimage in the method! I did that debug, I printed all the shaped and all the images of my dataset, however it doesn’t crash  But I know remembered that I didn’t try to iterate the dataloader, I only iterated the datasets. I am going to try that