code: def get_server_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--port", required=True, type=int, help="server port"
      )
      parser.add_argument(
          "--logdir",
          type=str,
          help="directory to save model/tensorboard data",
      )
      parser.add_argument(
          "--restore_model_path",
          type=str,
          help="restore model path for warm start",
      )
      parser.add_argument(
          "--restore_from_one_head",
          action="store_true",
          help="If set, will restore model from one head model. If ensemble_num > 1, will assign parameters of model0 to other models.",
      )
      parser.add_argument(
          "--restore_rpm_path",
          type=str,
          help="restore rpm path for warm start",
      )
      parser.add_argument(
          "--ensemble_num",
          type=int,
          required=True,
          help="model number to ensemble",
      )
      parser.add_argument(
          "--warm_start_batchs",
          type=int,
          default=100,
          help="collect how many batch data to warm start",
      )
      args = parser.parse_args()
      return args
  
label_desc: Use image node_id 1 for calling a global function with example usage: get_server_args() and returns: args