code: def _get_base_learning_rate(global_batch_size: int) -> float:
      """Returns the base learning rate for training 100 epochs with a given batch size.
  
      This follows section C.4 in https://arxiv.org/pdf/2105.04906.pdf.
  
      """
      if global_batch_size == 128:
          return 0.8
      elif global_batch_size == 256:
          return 0.5
      elif global_batch_size == 512:
          return 0.4
      else:
          return 0.3
  
imports: from typing import List, Tuple
  import torch
  from pytorch_lightning import LightningModule
  from torch import Tensor
  from torch.nn import Identity
  from torchvision.models import resnet50
  from lightly.loss.vicreg_loss import VICRegLoss
  from lightly.models.modules.heads import VICRegProjectionHead
  from lightly.models.utils import get_weight_decay_parameters
  from lightly.transforms.vicreg_transform import VICRegTransform
  from lightly.utils.benchmarking import OnlineLinearClassifier
  from lightly.utils.lars import LARS
  from lightly.utils.scheduler import CosineWarmupScheduler
label_desc: Use image node_id 1 for calling a global function with example usage: _get_base_learning_rate(global_batch_size) and returns: int, int, int, int