Solved by Nikronic in post #2 
              
              
                Hi, 
I do not know about x-shot learning but if you want to get a subset of your dataset, you just need to change the __len__() function to return maximum size you want. 
This function return the length of your dataset, so if you put a constant number or a function the current number, you get as muc…