self_supervised:自监督算法的Pytorch-Lightning实现-源码

自监督算法的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（默认配置）上使用