Shortcuts

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:
  • shift_factor (float) – The shift factor of cosine similarity. Default: 0.0.

  • scale_factor (float) – The scale factor of cosine similarity. Default: 1.0.

forward(pred, target, mask=None)[source]

Forward function of cosine similarity loss.

Parameters:
Returns:

The cosine similarity loss.

Return type:

torch.Tensor

Read the Docs v: latest
Versions
latest
stable
mmcls-1.x
mmcls-0.x
dev
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.