Skip to content

DenseCL, SSL model improvements

Compare
Choose a tag to compare
@MalteEbner MalteEbner released this 25 Jul 09:25
· 87 commits to master since this release
6e264d6
  • Adds the DenseCL: Dense Contrastive Learning for Self-Supervised Visual Pre-Training method. See the docs.
  • Add TiCoTransform, thanks @radiradev!
  • Improvements to the pre-commit hooks, thanks @SauravMaheshkar!
  • Fix memory bank issue when using gather_distributed=True and training on a single GPU
  • Fix student head update in DINO benchmark
  • Various improvements to MaskedVisionTransformer
  • Renaming of Lightly SSL to LightlySSL

Models