SimCLR(mlp, backbone=None, temperature=0.07)¶
A Simple Framework for Contrastive Learning of Visual Representations Details can be found from: https://arxiv.org/abs/2002.05709
x1 (torch.tensor) – a batch of image with augmentation. The input tensor shape should able to be feed into the backbone.
x2 (torch.tensor) – the size batch of image with different augmentation. The input tensor shape should able to be feed into the backbone.
- Return type