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)
Read more