code: def parse_args():
      parser = argparse.ArgumentParser(
          description="Convert the tf GPT-2 Model to Gluon."
      )
      parser.add_argument(
          "--tf_model_path",
          type=str,
          required=True,
          help="Directory of the tf GPT-2 model.",
      )
      parser.add_argument(
          "--save_dir",
          type=str,
          default=None,
          help="Directory path to save the converted GPT-2 model.",
      )
      parser.add_argument(
          "--test",
          action="store_true",
          help="Whether to test the conversion.",
      )
      return parser.parse_args()
  
imports: import os
  import re
  import json
  import shutil
  import logging
  import argparse
  import tensorflow
  from tensorflow.contrib.training import HParams
  from gpt_2.src import model
  import mxnet
  import numpy
  from numpy.testing import assert_allclose
  from gluonnlp.data.vocab import Vocab
  from gluonnlp.utils.misc import sha1sum, logging_config, naming_convention
  from gluonnlp.models.gpt2 import GPT2Model, GPT2ForLM
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: parser