After applying the transformation on your data (without training the model), you could save each sample via torch.save. Later in your training script, you could create a custom Dataset, as explained here 45, and load each augmented sample instead of the image.
Let me know, if you get stuck.