自监督学习库:solo-learn

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

提出了一种用于视觉表示学习的自监督方法库——自学习。该库使用Python实现,使用Pytorch和Pytorch lightning,通过具有混合精度的分布式培训管道、通过Nvidia DALI更快地加载数据、在线线性评估以获得更好的原型,以及许多其他培训技巧,满足了研究和行业的需求。目标是提供一个包含大量自我监督学习(Self-supervised Learning, SSL)方法的易用库,社区可以很容易地扩展和调整这些方法。单独学习为在廉价的小型基础设施上开发大预算的SSL解决方案开辟了途径,并通过让所有人都可以访问SSL来实现SSL的民主化。源代码可以在https://github.com/vturrisi/solo-learn上找到。

自监督学习库:solo-learn_第1张图片

你可能感兴趣的:(图神经网络,机器学习,科技,机器学习)