I need to shuffle a data. I do the train_test_split for ImageFolder() data manually and some classes do not fall into the train set. Set shuffle=True in DataLoader() is not a solution 
Is it a best solution to read all the images with numpy and then transform it to DataLoader objects? I think I’m messed up with data transfromations in pytorch.