code: def __init__(self, optim_args):
      self.optim_args = optim_args
      # Each parameter will get its own optimizer, which we keep track
      # of using this dictionary keyed on parameters.
      self.optim_objs = {}
  
imports: import random
  import warnings
  import weakref
  from collections import OrderedDict
  import numpy
  import torch
  from pyro.distributions import validation_enabled
label_desc: Use image node_id 1 to create a new Adam object  with example: obj = Adam(optim_args)