What would you want to know about data augmentation? How it is used in your code or how the underlying implementation is working?
In the former case, you could just use torchvision.transforms 11 to create your transformations and apply them in the Dataset.
In the latter case you might want to look at the source 10 to see the implementation. Let me know, if you need some more information.