code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--step",
          type=int,
          nargs="+",
          choices=(1, 2, 3),
          default=(1, 2, 3),
          help="Which steps of the ChatGPT example to run",
      )
      parser.add_argument(
          "--actor-model",
          type=lambda x: x.replace("facebook/opt-", ""),
          default="1.3b",
          choices=("1.3b", "6.7b", "13b", "66b"),
          help="Which facebook/opt-* model to use for Actor (step 1)",
      )
      parser.add_argument(
          "--reward-model",
          type=lambda x: x.replace("facebook/opt-", ""),
          default="350m",
          choices=("350m"),
          help="Which facebook/opt-* model to use for Reward (step 2)",
      )
      parser.add_argument(
          "--actor-zero-stage",
          type=str,
          default="",
          choices=("", "0", "1", "2", "3"),
          help="ZeRO stage for step 1 (Actor) training",
      )
      parser.add_argument(
          "--reward-zero-stage",
          type=str,
          default="",
          choices=("", "0", "1", "2", "3"),
          help="ZeRO stage for step 2 (Critic) training",
      )
      parser.add_argument(
          "--output-dir",
          type=lambda x: os.path.abspath(x),
          default="./output",
          help="Directory for output of each step",
      )
      parser.add_argument(
          "--deployment-type",
          type=str,
          default="single_gpu",
          choices=("single_gpu", "single_node", "multi_node"),
          help="Number of GPUs to run the actor/reward models on",
      )
      args = parser.parse_args()
  
      if args.actor_zero_stage != "" or args.reward_zero_stage != "":
          warnings.warn(
              "Non-default zero stages may result in OOM errors or worse performance."
          )
  
      return args
  
file_path: deepspeedexamples/applications/DeepSpeed-Chat/e2e_rlhf.py
imports: import argparse
  import warnings
  import subprocess
  import os
  import datetime
  import time
  import sys
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: args