code: def get_optimizer(opt):
      if opt == "sgd":
          return optim.SGD
      elif opt == "adam":
          return optim.Adam
      else:
          raise NotImplementedError
  
imports: import torch.nn
  import torch.optim
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: get_optimizer(opt) and returns: optim, optim