code: def parse_args():
      parser = argparse.ArgumentParser(
          description="Train a mxnet gluon model for FashonMNIST dataset"
      )
      parser.add_argument(
          "--batch-size", type=int, default=256, help="Batch size"
      )
      parser.add_argument(
          "--epochs", type=int, default=1, help="Number of Epochs"
      )
      parser.add_argument("--learning_rate", type=float, default=0.1)
      parser.add_argument(
          "--context",
          type=str,
          default="cpu",
          help="Context can be either cpu or gpu",
      )
      parser.add_argument(
          "--validate",
          type=bool,
          default=True,
          help="Run validation if running with smdebug",
      )
  
      opt = parser.parse_args()
      return opt
  
file_path: sagemaker-python-sdk/tests/data/mxnet_mnist/mnist_gluon.py
imports: import argparse
  import random
  import mxnet
  import numpy
  from mxnet import autograd, gluon
  from mxnet.gluon import nn
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: opt