You can create a Compose of augmentations and then use it in the training loop itslelf.
aug = Compose(<the list of augmentations>)

for x,y in dataloader:
    x_aug = aug(x)

I think this might do the trick.