/ssl_benchmarks_small

Implement some self-supervised learning frameworks on small datasets that easy to use.

Primary LanguagePython

Implement some self-supervised learning frameworks on small datasets that are easy to reuse.

  • Minimum dependency.
  • Easy to reproduce.

Todos

Cifar10 Cifar100 STL10
SimCLR ✔️ ✔️
BYOL
MoCo

Resnet18 backbone (Top1 accuracy)

Cifar10 Cifar100 STL10
SimCLR 91.76 65.86
BYOL 92.48 68.53
MoCo

Resnet50 backbone (Top1 accuracy)

Cifar10 Cifar100 STL10
SimCLR 93.23 67.43
BYOL
MoCo