code: def parse_args():
      parser = argparse.ArgumentParser(
          description="command for benchmark kv-store"
      )
      parser.add_argument(
          "--network",
          type=str,
          default="resnet",
          help="the neural network to test",
      )
      parser.add_argument(
          "--gpus",
          type=str,
          default="0,1",
          help='the gpus to be used, e.g "0,1,2,3"',
      )
      parser.add_argument(
          "--num-layers",
          type=int,
          default=152,
          help="number of layers, can be used for resnet",
      )
      parser.add_argument(
          "--kv-store",
          type=str,
          default="device",
          help="the kvstore type",
      )
      parser.add_argument(
          "--num-batches",
          type=int,
          default=5,
          help="number of batches to run",
      )
      parser.add_argument(
          "--disp-batches",
          type=int,
          default=1,
          help="show averaged results for every n batches",
      )
      parser.add_argument(
          "--test-results",
          type=int,
          default=1,
          help="if or not evalute the results correctness",
      )
      parser.add_argument(
          "--image-shape",
          type=str,
          default="3,224,224",
          help="input images shape",
      )
      parser.add_argument(
          "--num-classes",
          type=int,
          default=1000,
          help="number of classes",
      )
      parser.add_argument(
          "--optimizer",
          type=str,
          default="None",
          help="the optimizer set to kvstore. None means no optimizer",
      )
      parser.add_argument(
          "--gc-type",
          type=str,
          default="none",
          help="type of gradient compression",
      )
      args = parser.parse_args()
      logging.info(args)
      return args
  
file_path: mxnet/tools/bandwidth/measure.py
imports: import os, sys
  import mxnet
  import logging
  import argparse
  import time
  import numpy
  from importlib import import_module
  from collections import namedtuple
  from functools import reduce
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: args