code: def __init__(
      self, timestamp: int, device_type: DeviceType, bytes_: int
  ) -> None:
      self.pid = os.getpid()
      self.tid = threading.get_ident()
      self.timestamp = timestamp
      self.device_type = device_type
      self.device_id = (
          torch.cuda.current_device()
          if device_type == DeviceType.CUDA
          else -1
      )
      self.bytes = bytes_
  
imports: import os
  import threading
  import time
  from enum import Enum
  from typing import List
  import torch
  from colossalai.gemini.ophooks import BaseOpHook
  from colossalai.gemini.stateful_tensor import StatefulTensor
  from colossalai.legacy.engine import Engine
  from colossalai.legacy.utils.profiler.extention import ProfilerExtension
label_desc: Use image node_id 1 to create a new StatefulTensorMemoryEvent object  with example: obj = StatefulTensorMemoryEvent(timestamp, device_type, bytes_)