fix: Disabled mobilenet_v2 test for DLFW CI

Signed-off-by: Anurag Dixit <anuragd@nvidia.com>
pytorch · Nov 23, 2021 · 40c611f · 40c611f
1 parent a10613e
commit 40c611f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/py/test_api.py b/tests/py/test_api.py
@@ -508,7 +508,8 @@ def test_suite():
     suite = unittest.TestSuite()
     suite.addTest(unittest.makeSuite(TestLoggingAPIs))
     suite.addTest(TestCompile.parametrize(TestCompile, model=models.resnet18(pretrained=True)))
-    suite.addTest(TestCompile.parametrize(TestCompile, model=models.mobilenet_v2(pretrained=True)))
+    # Disabling mobilenet_v2 test due to https://nvbugs/3433655
+    # suite.addTest(TestCompile.parametrize(TestCompile, model=models.mobilenet_v2(pretrained=True)))
     suite.addTest(TestCompileHalf.parametrize(TestCompileHalf, model=models.resnet18(pretrained=True)))
     suite.addTest(TestCompileHalfDefault.parametrize(TestCompileHalfDefault, model=models.resnet18(pretrained=True)))
     suite.addTest(TestPTtoTRTtoPT.parametrize(TestPTtoTRTtoPT, model=models.resnet18(pretrained=True)))