Implement augmentations for our data

- Check what augmentation methods were used by SimCLR
- Think of a list of augmentation techniques that makes sense for CT data. We will discuss it together
- After that, we need to implement some of these techniques and integrate them into our training pipeline