code: def setUp(self):
      self.resnet_variants = dict(
          resnet18=dict(
              num_ftrs=512,
              proj_hidden_dim=512,
              pred_hidden_dim=128,
              out_dim=512,
          ),
          resnet50=dict(
              num_ftrs=2048,
              proj_hidden_dim=2048,
              pred_hidden_dim=512,
              out_dim=2048,
          ),
      )
      self.batch_size = 2
      self.input_tensor = torch.rand((self.batch_size, 3, 32, 32))
  
file_path: lightly/tests/models/test_ModelsSimSiam.py
imports: import unittest
  import torch
  import torch.nn
  import torchvision
  from lightly.models import SimSiam
label_desc: Use image node_id 1 for calling the TestSimSiam obj's underlying member method code with example usage: obj.setUp() without return types