https://www.omegaxyz.com/2022/02/02/contrastive_learning_data_augmentation/