code: def parse_config():
      parser = argparse.ArgumentParser(description="arg parser")
      parser.add_argument(
          "--checkpoint_path",
          type=str,
          default="/dataset/models/checkpoint-1000",
      )
      parser.add_argument(
          "--trainable_params", type=str, default="embed,norm"
      )
      args = parser.parse_args()
      return args
  
imports: import os
  import torch
  import argparse
label_desc: Use image node_id 1 for calling a global function with example usage: parse_config() and returns: args