code: def __init__(self, vit):
      super().__init__()
  
      self.mask_ratio = 0.15
      self.backbone = MAEBackbone.from_vit(vit)
      self.projection_head = MSNProjectionHead(384)
  
      self.anchor_backbone = copy.deepcopy(self.backbone)
      self.anchor_projection_head = copy.deepcopy(self.projection_head)
  
      utils.deactivate_requires_grad(self.backbone)
      utils.deactivate_requires_grad(self.projection_head)
  
      self.prototypes = nn.Linear(256, 1024, bias=False).weight
  
file_path: lightly/examples/pytorch/pmsn.py
imports: import copy
  import torch
  import torchvision
  from torch import nn
  from lightly.loss import PMSNLoss
  from lightly.models import utils
  from lightly.models.modules.heads import MSNProjectionHead
  from lightly.models.modules.masked_autoencoder import MAEBackbone
  from lightly.transforms import MSNTransform
label_desc: Use image node_id 1 to create a new PMSN object from inherited base classes: nn with example: obj = PMSN(vit)