Shortcuts

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

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.