Stratified_sampler_fastai https://medium.com/@champs.jaideep/stratified-batch-sampler-using-fastai-20f9b898609d?source=friends_link&sk=66ee9b03077e979f069390abf4d31c5a please go through the medium link to understand the details.