Skip to content

Latest commit

 

History

History
12 lines (8 loc) · 512 Bytes

README.md

File metadata and controls

12 lines (8 loc) · 512 Bytes

SimCLRv2 PyTorch

simclr_anim

A PyTorch implementation of SimCLRv2 architecture developed by Google Research. This code is heavily based off Spijkervet SimCLRv1 implementation in PyTorch. ResNet architecture and contrastive head are borrowed from Separius' excellent repo for converting Tensorflow weights to PyTorch.

Experimental.