thanks @Deeply! Oversampling for me is done through augmentation using the transforms class. Can you elaborate what you mean on “randomly sample”? Do you know of a code example of how to randomly sample the lower size dataset?