code: def make_optimizer(args, model):
      param_group = {
          "params": [p for p in model.parameters() if p is not None]
      }
  
      optimizer = flow.optim.SGD(
          [param_group],
          lr=args.lr,
          momentum=args.momentum,
          weight_decay=args.weight_decay,
      )
      return optimizer
  
imports: import oneflow
  from oneflow.nn.parallel import DistributedDataParallel
  from utils.ofrecord_data_utils import OFRecordDataLoader, SyntheticDataLoader
  from utils.utils_logging import AverageMeter
  from utils.utils_callbacks import CallBackVerification, CallBackLogging, CallBackModelCheckpoint
  from backbones import get_model
  from graph import TrainGraph, EvalGraph
  from utils.losses import CrossEntropyLoss_sbp
  import logging
label_desc: Use image node_id 2 for calling a global function with example usage: make_optimizer(args, model) and returns: optimizer