/simclr-pytorch

PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers