This way i’d call it alteration, not augmentation. Augmentation is when you are creating additional training samples. You need to move transformations to init, transform all x’es and add result to original data.
Also take a look at timm 1 library for the augmentations, cutmix and mixup implementations helped me a lot in recent project.