I’m trying to explore using different size of per-class samples for transfer learning. For example, I want to only take 10 images per class for training versus 50 images per class. How do I specify the number of per-class samples without reorganizing the dataset?