code: def get_sizes(output, type):
      match = re.findall(
          type + r": +([\d,\.]+) +([\d,\.]+) +([\d,\.]+) MB", output
      )
      (consumed, peaked, total) = map(str2flt, match[0])
      return consumed, peaked, total
  
imports: import numpy
  import torch
  import re, gc
  from math import isclose
  import inspect
label_desc: Use image node_id 17 for calling a global function with example usage: get_sizes(output, type) and returns: consumed,  peaked,  total