Skip to content

Commit

Permalink
Add reference in siglip docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
rwightman committed Sep 18, 2023
1 parent 134e61a commit 8ba778d
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion src/open_clip/loss.py
Original file line number Diff line number Diff line change
Expand Up @@ -305,7 +305,15 @@ def neighbour_exchange_bidir_with_grad(left_rank, right_rank, tensor_to_left, te


class SigLipLoss(nn.Module):

""" Sigmoid Loss for Language Image Pre-Training (SigLIP) - https://arxiv.org/abs/2303.15343
@article{zhai2023sigmoid,
title={Sigmoid loss for language image pre-training},
author={Zhai, Xiaohua and Mustafa, Basil and Kolesnikov, Alexander and Beyer, Lucas},
journal={arXiv preprint arXiv:2303.15343},
year={2023}
}
"""
def __init__(
self,
cache_labels=False,
Expand Down

0 comments on commit 8ba778d

Please sign in to comment.