SimCLR-pytorch
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)
Contributions
If you have any suggestions or hiccups, feel free to raise a PR or Issue. All contributions welcome!
License
GitHub
GitHub - rish-16/SimCLR-pytorch at pythonawesome.com
Unofficial PyTorch implementation of SimCLR by Google Brain - GitHub - rish-16/SimCLR-pytorch at pythonawesome.com
rish-16
