btw do you know if the answer u provided can aid in solving:

  
    
    
    How does one create a data set in pytorch and save it into a file to later be used? 
  
  
    I want to extract data from cifar10 in a specific order according to some criterion f(Image,label) (for the sake of an example lets say f(Image,label) simply computes the sum of all the pixels in Image). Then it I want to generate 1 file for the train set and 1 file for the test set that I can later load in a dataloader to use for training a neural net. 
How do I do this? My current idea was simply to loop through the data with data loader with shuffle off and remember the indices of the images …
  


thats really the problem I am trying to solve.