Published March 16, 2022 | Version v1
Journal article Open

solo-learn: A Library of Self-supervised Methods for Visual Representation Learning

  • 1. University of Trento, Italy
  • 2. SAP AI Research - Berlin, Germany

Description

This paper presents solo-learn, a library of self-supervised methods for visual rep-
resentation learning. Implemented in Python, using Pytorch and Pytorch lightning, the
library ts both research and industry needs by featuring distributed training pipelines
with mixed-precision, faster data loading via Nvidia DALI, online linear evaluation for
better prototyping, and many additional training tricks. Our goal is to provide an easy-to-
use library comprising a large amount of Self-supervised Learning (SSL) methods, that
can be easily extended and ne-tuned by the community. solo-learn opens up av-
enues for exploiting large-budget SSL solutions on inexpensive smaller infrastructures and
seeks to democratize SSL by making it accessible to all. The source code is available at
https://github.com/vturrisi/solo-learn.

Files

21-1155 (2).pdf

Files (306.1 kB)

Name Size Download all
md5:7569d191ab38f1783745519beee409b1
306.1 kB Preview Download

Additional details

Funding

AI4Media – A European Excellence Centre for Media, Society and Democracy 951911
European Commission