Unofficial PyTorch implementation of SimCLR by Google Brain

Unofficial PyTorch implementation of SimCLR by Google Brain [abs
, pdf
]
Installation
You can install the package via pip
:
pip install simclr_pytorch
Usage
You can use the SimCLR
model like so:
from simclr_pytorch import SimCLR
train_loader = ...
bigcnn = ResNet50(...)
simclr = SimCLR(
image_size=(32, 32),
bigcnn=bigcnn
)
trained_bigcnn = simclr.fit(train_loader)