CosineSimilarityLoss¶
- class mmpretrain.models.losses.CosineSimilarityLoss(shift_factor=0.0, scale_factor=1.0)[source]¶
Cosine similarity loss function.
Compute the similarity between two features and optimize that similarity as loss.
- Parameters:
- forward(pred, target, mask=None)[source]¶
Forward function of cosine similarity loss.
- Parameters:
pred (torch.Tensor) – The predicted features.
target (torch.Tensor) – The target features.
- Returns:
The cosine similarity loss.
- Return type: