The number of images per class to use during training will now be capped to the number of images in the dataset