MixMIMPretrainHead¶
- class mmpretrain.models.heads.MixMIMPretrainHead(loss, norm_pix=False, patch_size=16)[源代码]¶
Head for MixMIM Pre-training.
- 参数:
- loss(x_rec, target, mask)[源代码]¶
Generate loss.
- 参数:
pred (torch.Tensor) – The reconstructed image.
target (torch.Tensor) – The target image.
mask (torch.Tensor) – The mask of the target image.
- 返回:
The reconstruction loss.
- 返回类型: