https://github.com/earthspecies/representation-learning/blob/d13e0e88c05ae56d18940f2a2c5a7d15f7db7a00/representation_learning/data/dataset.py#L243 I think this applies augmentations (e.g. mixup) to test and val data, which is not typical