
Barlow Twins框架的实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
《Barlow Twins框架的实现》简介:本文详细介绍了双胞胎网络架构Barlow Twins的设计理念与技术细节,探索了其在自监督学习领域的应用价值及实验结果。
Barlow Twins的PyTorch实施论文正在进行中。该代码是SimSiam实现的一个修改版本,在V100 GPU上每个纪元的时间约为40秒,GPU使用量约为9 GB。
当前版本在测试集上的准确率达到84.7%左右。接下来的工作包括:
- 从零开始的热身学习率
- 报告关于CIFAR-10的结果
- 创建公共接口以方便添加
安装依赖项,请执行以下命令:
```
pip install -r requirements.txt
```
所需的库包括:PyTorch、PyTorch Lightning、TorchVision和Lightning。
我们遵循KNN评估协议,在CIFAR-10数据集上对Barlow Twins模型进行基准测试。目前,该模型的最高准确率为84.7%。
全部评论 (0)
还没有任何评论哟~


