code: def draw_graph(model, args):
      if args is None:
          args = [1, 3, 224, 224]  # assume ImageNet default
  
      # if args is not Tensor but is array like then convert it to torch tensor
      if (
          not isinstance(args, torch.Tensor)
          and hasattr(args, "__len__")
          and hasattr(args, "__getitem__")
          and not isinstance(args, (str, abc.ByteString))
      ):
          args = torch.ones(args)
  
      dot = draw_img_classifier(model, args)
      return DotWrapper(dot)
  
file_path: tensorwatch/tensorwatch/model_graph/hiddenlayer/pytorch_draw_model.py
imports: import pydot
  from .summary_graph import SummaryGraph
  from .None import distiller
  import torch
  import collections.abc
  import os
label_desc: Use image node_id 1 for calling a global function with example usage: draw_graph(model, args) and returns: DotWrapper