text: # model settings
  model = dict(
      type='BarlowTwins',
      backbone=dict(
          type='ResNet',
          depth=50,
          in_channels=3,
          out_indices=[4],  # 0: conv-1, x: stage-x
          norm_cfg=dict(type='SyncBN'),
          zero_init_residual=True),
      neck=dict(
          type='NonLinearNeck',
          in_channels=2048,
          hid_channels=8192,
          out_channels=8192,
          num_layers=3,
          with_last_bn=False,
          with_last_bn_affine=False,
          with_avg_pool=True,
          init_cfg=dict(
              type='Kaiming', distribution='uniform', layer=['Linear'])),
      head=dict(type='LatentCrossCorrelationHead', in_channels=8192))
  