Unofficial PyTorch implementation of SimCLR by Google Brain [abs, pdf]
You can install the package via pip:
pip install simclr_pytorchYou 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)If you have any suggestions or hiccups, feel free to raise a PR or Issue. All contributions welcome!