Shortcuts

MixMIM

class mmpretrain.models.selfsup.MixMIM(backbone, neck=None, head=None, pretrained=None, data_preprocessor=None, init_cfg=None)[source]

MixMIM.

Implementation of MixMIM: Mixed and Masked Image Modeling for Efficient Visual Representation Learning..

loss(inputs, data_samples, **kwargs)[source]

The forward function in training.

Parameters:
  • inputs (torch.Tensor) – The input images.

  • data_samples (List[DataSample]) – All elements required during the forward function.

Returns:

A dictionary of loss components.

Return type:

Dict[str, 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.