文件名称:barlowtwins:Barlow Twins纸的实施
文件大小:65KB
文件格式:ZIP
更新时间:2024-04-01 10:27:43
self-supervised-learning Python
barlowtwins Barlow Twins的PyTorch实施论文: 目前这项工作正在进行中。 该代码是SimSiam实现的修改版本 在V100 GPU上,每个纪元的时间约为40秒 GPU使用量约为9 GB 当前版本的测试准确性达到84.7%左右 去做: 从0开始的热身学习率 报告关于cifar-10的结果 创建公关以轻松添加 安装 pip install -r requirements.txt 依存关系 火炬 PyTorch闪电 火炬视觉 轻轻 基准测试 我们遵循KNN评估协议,在CIFAR-10数据集上对BarlowTwins模型进行基准测试。 目前,尽力而为达到了84.7%的测试准确性。 准确性 失利 纸
【文件预览】:
barlowtwins-main
----.gitignore(31B)
----requirements.txt(21B)
----utils.py(4KB)
----README.md(1KB)
----loss.py(867B)
----main.py(4KB)
----docs()
--------loss_logs.png(36KB)
--------accuracy_logs.png(35KB)