SwAVHead¶ class mmpretrain.models.heads.SwAVHead(loss)[源代码]¶ Head for SwAV Pre-training. 参数: loss (dict) – Config dict for module of loss functions. loss(pred)[源代码]¶ Generate loss. 参数: pred (torch.Tensor) – NxC input features. 返回: The SwAV loss. 返回类型: torch.Tensor