code: def setUp(self):
      self.resnet_variants = ["resnet-18", "resnet-50"]
      self.batch_size = 2
      self.input_tensor = torch.rand((self.batch_size, 3, 32, 32))
  
file_path: lightly/tests/models/test_ModelsBYOL.py
imports: import unittest
  import torch
  import torch.nn
  import torchvision
  import lightly
  from lightly.models import BYOL, ResNetGenerator
label_desc: Use image node_id 1 for calling the TestModelsBYOL obj's underlying member method code with example usage: obj.setUp() without return types