Solo-learn：一个用于视觉表征学习的自监督方法库
计算机视觉与模式识别
2022-02-07 v4
摘要
本文介绍 solo-learn，一个用于视觉表征学习的自监督方法库。该库使用 Python 实现，基于 Pytorch 和 Pytorch lightning，通过配备混合精度分布式训练流水线、借助 Nvidia DALI 实现更快数据加载、在线线性评估以便于原型设计以及众多额外的训练技巧，同时满足研究和工业需求。我们的目标是提供一个易于使用的库，包含大量自监督学习（SSL）方法，并可由社区轻松扩展和微调。solo-learn 为在廉价的小型基础设施上利用大规模预算的 SSL 解决方案开辟了途径，并致力于通过使 SSL 人人可及来实现其大众化。源代码可在 https://github.com/vturrisi/solo-learn 获取。
引用
@article{arxiv.2108.01775,
title = {Solo-learn: A Library of Self-supervised Methods for Visual Representation Learning},
author = {Victor G. Turrisi da Costa and Enrico Fini and Moin Nabi and Nicu Sebe and Elisa Ricci},
journal= {arXiv preprint arXiv:2108.01775},
year = {2022}
}
备注
Accepted to JMLR