pytorchvideo.models.simclr¶
-
class
pytorchvideo.models.simclr.
SimCLR
(mlp, backbone=None, temperature=0.07)[source]¶ A Simple Framework for Contrastive Learning of Visual Representations Details can be found from: https://arxiv.org/abs/2002.05709
-
forward
(x1, x2)[source]¶ - Parameters
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
-