code: def colo_model_optimizer_usage(optim) -> Tuple[int, int]:
      """Trace the optimizer memory usage
  
      Args:
          optim (ShardedOptimV2): an instance of ShardedOptimizer
  
      Returns:
          Tuple[int, int]: cuda/cpu memory usage in Byte
      """
      if optim is None:
          return 0, 0
      assert hasattr(
          optim, "get_memory_usage"
      ), f"{type(optim)} has no attr get_memory_usage()"
      return optim.get_memory_usage()
  
label_desc: Use image node_id 1 for calling a global function with example usage: colo_model_optimizer_usage(optim) and returns: optim, int,  int