contrastive learning notebook