https://github.com/VisionLearningGroup/OP_Match/blob/fd55de2e1913ffbcb3fd413bd9f024afce44c024/dataset/cifar.py#L60