If you're interested in using another data augmentation library, learn how in the Albumentations or Kornia notebooks.