chore(//tests): Adding instructions on where to get model for PTQ python

test

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

tests/py/test_ptq_dataloader_calibrator.py

@@ -81,6 +81,8 @@ def test_compile_script(self):
 
 def test_suite():
     suite = unittest.TestSuite()
+    # You need a pre-trained VGG cifar10 model to run this test. Please follow instructions at
+# https://github.com/NVIDIA/TRTorch/tree/master/cpp/ptq/training/vgg16 to export this model.
     suite.addTest(TestAccuracy.parametrize(TestAccuracy, model=torch.jit.load('./trained_vgg16.jit.pt')))
 
     return suite

tests/py/test_ptq_to_backend.py

@@ -84,6 +84,8 @@ def test_compile_script(self):
 
 def test_suite():
     suite = unittest.TestSuite()
+    # You need a pre-trained VGG cifar10 model to run this test. Please follow instructions at
+    # https://github.com/NVIDIA/TRTorch/tree/master/cpp/ptq/training/vgg16 to export this model.
     suite.addTest(TestAccuracy.parametrize(TestAccuracy, model=torch.jit.load('./trained_vgg16.jit.pt')))
 
     return suite