We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 266a96c commit 789a602Copy full SHA for 789a602
instance_loss.py
@@ -16,7 +16,7 @@ def __init__(self, gamma = 1) -> None:
16
17
def forward(self, feature, label = None) -> Tensor:
18
# Dual-Path Convolutional Image-Text Embeddings with Instance Loss, ACM TOMM 2020
19
- # https://arxiv.org/abs/1711.05535
+ # https://zdzheng.xyz/files/TOMM20.pdf
20
# using cross-entropy loss for every sample if label is not available. else use given label.
21
normed_feature = l2_norm(feature)
22
sim1 = torch.mm(normed_feature*self.gamma, torch.t(normed_feature))
0 commit comments