chore: use rn18 instead of rn50

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

tests/cpp/test_modules_as_engines.cpp

@@ -24,7 +24,7 @@ INSTANTIATE_TEST_SUITE_P(
     ModuleAsEngineForwardIsCloseSuite,
     CppAPITests,
     testing::Values(
-        PathAndInput({"tests/modules/resnet50_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}, 0.99}),
+        PathAndInput({"tests/modules/resnet18_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}, 0.99}),
         PathAndInput({"tests/modules/mobilenet_v2_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}, 0.99}),
         PathAndInput({"tests/modules/efficientnet_b0_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}, 0.99}),
         PathAndInput({"tests/modules/vit_scripted.jit.pt", {{1, 3, 224, 224}}, {at::kFloat}, 0.99})));

tests/py/models/test_models.py

@@ -10,8 +10,8 @@
 
 
 class TestModels(unittest.TestCase):
-    def test_resnet50(self):
-        self.model = models.resnet50(pretrained=True).eval().to("cuda")
+    def test_resnet18(self):
+        self.model = models.resnet18(pretrained=True).eval().to("cuda")
         self.input = torch.randn((1, 3, 224, 224)).to("cuda")
 
         compile_spec = {
@@ -120,8 +120,8 @@ def test_bert_base_uncased(self):
                 msg=f"HF BERT base-uncased TRT outputs don't match with the original model. Cosine sim score: {cos_sim} Threshold: {COSINE_THRESHOLD}",
             )
 
-    def test_resnet50_half(self):
-        self.model = models.resnet50(pretrained=True).eval().to("cuda")
+    def test_resnet18_half(self):
+        self.model = models.resnet18(pretrained=True).eval().to("cuda")
         self.input = torch.randn((1, 3, 224, 224)).to("cuda")
         self.scripted_model = torch.jit.script(self.model)
         self.scripted_model.half()