-
Pytorch-Lightning实现的自监督算法：self_supervised
资源介绍
自监督算法的Pytorch-Lightning实现
这是一个实现，和使用。可以对配置进行调整，以实现一系列可能的自我监督实现。
我们最近添加了必要的功能来运行和进行比较。
有关更多详细信息，请参见博客文章。
安装
确保你在一个新的很conda或venv环境中，然后运行：
git clone https://github.com/untitled-ai/self_supervised
cd self_supervised
pip install -r requirements.txt
复制我们的结果
您可以通过运行python train_blog.py复制我们博客文章的结果。 z和z'之间的余弦相似度报告为step_neg_cos （对于负面示例）和step_pos_cos （对于正面示例）。分类准确性报告为valid_class_acc 。
入门
要开始在STL-10（默认配置）上使用