ssl-cifar10

Simple transfer learning on cifar10 with self-supervised backbone SwAV using Pytorch lightning and bolts

Install requirements : pip install -r requirements.txt

Start training on CIFAR10 : python model.py