Interesting to see contrastive loss instead of a reconstruction loss.