code: def __init__(self, model):
      # sanity check.
      super(PytorchGraph, self).__init__(model)
      self.model = model
      self.state_dict = _unique_state_dict(self.model)
      self.shape_dict = dict()
      self.layer_weight_map = dict()
  
imports: from mmdnn.conversion.common.DataStructure.graph import GraphNode, Graph
  import torch
  import torch.autograd
  import torch.serialization
  import contextlib
  from torch.jit import _unique_state_dict
label_desc: Use image node_id 1 to create a new PytorchGraph object from inherited base classes: Graph with example: obj = PytorchGraph(model)