/sslgym

Implementations of various Self Supervised Learning methods

Primary LanguagePythonMIT LicenseMIT

Implementation of various Self Supervised Learning methods in various frameworks

Build and Tests pre-commit.ci status

This repository aims to contain Tensorflow and PyTorch Implementations of various Self Supervised Learning methods.

At the moment, implementations are provided for:-

  • DINO
  • SwAV
  • VICReg