LatentCrossCorrelationHead¶
- class mmpretrain.models.heads.LatentCrossCorrelationHead(in_channels, loss, init_cfg=None)[source]¶
Head for latent feature cross correlation.
Part of the code is borrowed from script.
- Parameters:
- loss(input, target)[source]¶
Forward head.
- Parameters:
input (torch.Tensor) – NxC input features.
target (torch.Tensor) – NxC target features.
- Returns:
The cross correlation loss.
- Return type: