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