
对比学习方法,其源代码。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
对比学习方法能够支持更广泛的内容进行第三方PyTorch实现,具体请参考“可用内容”部分。为了探索其潜力,我们利用SimCLR进行对比学习的预训练过程,并结合停止梯度在线线性评估技术,充分利用Pytorch闪电登录和默认收益(包括多GPU训练以及混合精度等优化)。此外,我们还在GPU设备上收集负样本,以模拟更大的批次大小,尽管这些负样本的梯度不会在GPU上更新。为了进一步提升数据加载速度,我们采取了增加GPU内存使用量的策略。SimCLR的多分辨率图像处理能力也得到了应用。同时,SimCLR与预训练后线性评估相结合,通常能够获得1-1.5%的准确度提升。该方法在结果模型、方法、数据集、时代、批温以及大规模监督下都表现出色,尤其在在线线性评估和预训练后线性评估方面展现出优势。具体而言,Resnet18与SimCLR在Imagenet-100数据集上,使用100、256的参数规模和0.2的批温时分别取得了70.74和71.02的准确率。
全部评论 (0)
还没有任何评论哟~


