contrastive loss 的高级代码实现 (pytorch)

耐心是关键

于 2021-08-17 10:55:46 发布

阅读量4.6k

点赞数 3

分类专栏： PyTorch 深度学习

本文链接：https://blog.csdn.net/guotong1988/article/details/119751692

版权

深度学习同时被 2 个专栏收录

216 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

PyTorch

92 篇文章 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

该博客深入探讨了Contrastive Loss的概念，并提供了一个基于PyTorch的SimCLRLoss高级代码实现，用于 Sentence Transformers 模型的训练。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

    LARGE_NUM = 1e9
    def _contrastive_loss_forward(self,
                                  hidden1: torch.Tensor,
                                  hidden2: torch.Tensor,
                                  hidden_norm: bool

了解本专栏

超级会员免费看