Shortcuts

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

torch.Tensor

Read the Docs v: latest
Versions
latest
stable
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.