new_contents: import unittest
  import logging
  from .helpers import run_module
  
  logging.basicConfig(level=logging.DEBUG)
  
  
  class TorchTests(unittest.TestCase):
      def setUp(self):
          self.name = "benchmarker"
          self.imgnet_args = [
              "--framework=pytorch",
              "--problem_size=4",
              "--batch_size=2",
          ]
  
      def test_vgg16(self):
          run_module(self.name, "--problem=vgg16", *self.imgnet_args)
  
      def test_resnet50(self):
          run_module(self.name, "--problem=resnet50", *self.imgnet_args)
  
subject: Add pytorch tests