CrossCorrelationLoss¶
- class mmpretrain.models.losses.CrossCorrelationLoss(lambd=0.0051)[source]¶
Cross correlation loss function.
Compute the on-diagnal and off-diagnal loss.
- Parameters:
lambd (float) – The weight for the off-diag loss.
- forward(cross_correlation_matrix)[source]¶
Forward function of cross correlation loss.
- Parameters:
cross_correlation_matrix (torch.Tensor) – The cross correlation matrix.
- Returns:
cross correlation loss.
- Return type: