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)