Closed Sherrylone closed 4 years ago
Hi. Thanks for your great work. But I have a little confusion. You implement the contrastive loss by Cross-Entropy Loss without softmax function. So the negatives actually didn't work but only positives.
@Sherrylone The softmax function is applied in torch.nn.CrossEntropyLoss.
torch.nn.CrossEntropyLoss
Hi. Thanks for your great work. But I have a little confusion. You implement the contrastive loss by Cross-Entropy Loss without softmax function. So the negatives actually didn't work but only positives.