Contrastive Learning Pytorch