code: def __init__(self):
      super().__init__()
      resnet = torchvision.models.resnet18()
      self.backbone = nn.Sequential(*list(resnet.children())[:-2])
      self.projection_head = BarlowTwinsProjectionHead(512, 2048, 2048)
      self.local_projection_head = VicRegLLocalProjectionHead(
          512, 128, 128
      )
      self.average_pool = nn.AdaptiveAvgPool2d(output_size=(1, 1))
      self.criterion = VICRegLLoss()
  
file_path: lightly/examples/pytorch_lightning_distributed/vicregl.py
imports: import pytorch_lightning
  import torch
  import torchvision
  from torch import nn
  from lightly.loss import VICRegLLoss
  from lightly.models.modules import BarlowTwinsProjectionHead
  from lightly.models.modules.heads import VicRegLLocalProjectionHead
  from lightly.transforms.vicregl_transform import VICRegLTransform
label_desc: Use image node_id 1 to create a new VICRegL object from inherited base classes: pl with example: obj = VICRegL()