Skip to content

Latest commit

 

History

History
16 lines (10 loc) · 276 Bytes

README.md

File metadata and controls

16 lines (10 loc) · 276 Bytes

SimCLR v2 PyTorch

SimCLRv2 Architecture

Train with CIFAR 10

pip3 install pytorch-lightning

python3 train_cifar10.py

To-Do

  • Support more datasets
  • Supervised fine-tuning
  • Distillation