code: def test_graph(self):
      model = torchvision.models.vgg16()
      g = hl.build_graph(model, torch.zeros([1, 3, 224, 224]))
      g.save(os.path.join(OUTPUT_DIR, "pytorch_vgg16.pdf"))
  
      model = torchvision.models.resnet50()
      g = hl.build_graph(model, torch.zeros([1, 3, 224, 224]))
      g.save(os.path.join(OUTPUT_DIR, "pytorch_resnet50.pdf"))
  
      # Clean up
      shutil.rmtree(OUTPUT_DIR)
  
file_path: hiddenlayer/tests/test_pytorch_graph.py
imports: import os
  import sys
  import shutil
  import unittest
  import torch
  import torchvision.models
  import hiddenlayer
  from hiddenlayer import transforms
label_desc: Use image node_id 1 for calling the TestPytorchGraph obj's underlying member method code with example usage: obj.test_graph() without return types