code: def __init__(self, name) -> None:
      self.name = name
      self.start_time = 0.0
      self.end_time = 0.0
      self.inference_memory = 0
      self.input_shape: Sequence[int] = []
      self.output_shape: Sequence[int] = []
      self.MAdd = 0
      self.duration = 0.0
      self.Flops = 0
      self.Memory = 0, 0
      self.parameter_quantity = 0
      self.done = False
  
imports: import time
  from collections import OrderedDict
  from typing import Dict, Sequence
  import functools
  import itertools
  import numpy
  import torch
  import torch.nn
  from .compute_madd import compute_madd
  from .compute_flops import compute_flops
  from .compute_memory import compute_memory
  from .stat_tree import StatTree, StatNode
  from .reporter import report_format
label_desc: Use image node_id 1 to create a new ModuleStats object  with example: obj = ModuleStats(name)