code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--device",
          type=int,
          default=0,
          help="GPU to allocate to pytorch for model, graph data, and node label storage",
          required=False,
      )
  
      parser.add_argument(
          "--features_device",
          type=str,
          default="0",
          help="Device to allocate to pytorch for feature storage",
          required=False,
      )
  
      parser.add_argument(
          "--num_epochs",
          type=int,
          default=1,
          help="Number of training epochs",
          required=False,
      )
  
      return parser.parse_args()
  
file_path: cugraph/python/cugraph-pyg/cugraph_pyg/examples/graph_sage_sg.py
imports: import time
  import argparse
  import gc
  import torch
  from torch_geometric.nn import CuGraphSAGEConv
  import torch.nn
  import torch.nn.functional
  from typing import Union
label_desc: Use image node_id 3 for calling a global function with example usage: parse_args() and returns: parser