class_docstr: Helper class to create graphs to display memory
class_docstr_tok: ["Helper","class","to","create","graphs","to","display","memory"]
code: def __init__(self):
      import matplotlib
  
      self.font = {
          "family": matplotlib.rcParams["font.family"],
          "weight": "normal",
          "size": 12,
      }
  
imports: from contextlib import contextmanager
  from dataclasses import dataclass
  from enum import auto, Enum
  from functools import lru_cache
  from typing import Dict, List, NamedTuple, Tuple, Union
  import numpy
  import torch
  import torch.nn
  from fairscale.nn import FullyShardedDataParallel
  from vissl.utils.visualize import matplotlib_figure_to_image
label_desc: Use image node_id 1 to create a new MemoryGraphCreator object  with example: obj = MemoryGraphCreator()