I think this suggestion by @ptrblck would work for me. I’m relatively new to Pytorch and not exactly sure what it would look like to iterate the dataset and store the augmented tensors. @Sameer_Verma would you be willing to post the code you used to do this?