InfoNCE 对比学习损失

大模型手撕 · 中等 · 损失函数、对比学习、Embedding

实现 batch 内负样本的 InfoNCE 损失(CLIP / SimCSE 风格的单向版本):

数据范围

在 RagCodes 上用 Python / C++ / Java 在线编写并提交,核心代码模式与 ACM 模式均可评测。