code: def resnet_generator(name: str):
      if name == "resnet18":
          return torchvision.models.resnet18()
      elif name == "resnet50":
          return torchvision.models.resnet50()
      raise NotImplementedError
  
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 a global function with example usage: resnet_generator(name) and returns: torchvision, torchvision