code: def __init__(self, backbone):
      super().__init__()
      self.backbone = backbone
      self.projection_head = SwaVProjectionHead(512, 512, 128)
      self.prototypes = SwaVPrototypes(128, 512, 1)
  
      self.start_queue_at_epoch = 2
      self.queues = nn.ModuleList(
          [MemoryBankModule(size=(3840, 128)) for _ in range(2)]
      )
  
file_path: lightly/examples/pytorch/swav_queue.py
imports: import torch
  import torchvision
  from torch import nn
  from lightly.loss import SwaVLoss
  from lightly.models.modules import SwaVProjectionHead, SwaVPrototypes
  from lightly.models.modules.memory_bank import MemoryBankModule
  from lightly.transforms.swav_transform import SwaVTransform
label_desc: Use image node_id 1 to create a new SwaV object from inherited base classes: nn with example: obj = SwaV(backbone)