
SupContrast: PyTorch中的“监督对比学习”实现(附SimCLR)
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本文介绍了SupContrast项目,它是在PyTorch框架下实现的监督对比学习代码库,并提供了与SimCLR的相关比较。
SupContrast:监督式对比学习
此库使用CIFAR作为示例来展示以下论文在PyTorch中的实现:
1. 监督式对比学习。
2. 视觉表示的简单框架进行对比学习。
损失函数
损耗函数位于losses.py文件中,它接收features(L2归一化)和labels作为输入,并返回损耗。如果未提供labels,则会退化为SimCLR的形式。
用法:
```python
from losses import SupConLoss
# 定义带有温度参数`temp`的损失函数
criterion = SupConLoss(temperature=temp)
# features: [bsz, n_views, f_dim]
# `n_views`是每张图像的不同裁剪数量
# 建议对features进行L2归一化处理。
```
全部评论 (0)
还没有任何评论哟~


