code: def main():
      args = parse_args()
      process_checkpoint(args.in_file, args.out_file, args.save_keys)
  
imports: import argparse
  import subprocess
  import torch
  from mmengine.logging import print_log
  from mmengine.utils import digit_version
label_desc: Use image node_id 3 for calling a global function with example usage: main() without return types