Yes, I have already preloaded the complete dataset.
And just now I had finished preparing the dataset through your advice.
Thank you for your answer to the discussion:https://discuss.pytorch.org/t/input-numpy-ndarray-instead-of-images-in-a-cnn/18797.I 123 learned a lot.