File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -508,7 +508,8 @@ def test_suite():
508
508
suite = unittest .TestSuite ()
509
509
suite .addTest (unittest .makeSuite (TestLoggingAPIs ))
510
510
suite .addTest (TestCompile .parametrize (TestCompile , model = models .resnet18 (pretrained = True )))
511
- suite .addTest (TestCompile .parametrize (TestCompile , model = models .mobilenet_v2 (pretrained = True )))
511
+ # Disabling mobilenet_v2 test due to https://nvbugs/3433655
512
+ # suite.addTest(TestCompile.parametrize(TestCompile, model=models.mobilenet_v2(pretrained=True)))
512
513
suite .addTest (TestCompileHalf .parametrize (TestCompileHalf , model = models .resnet18 (pretrained = True )))
513
514
suite .addTest (TestCompileHalfDefault .parametrize (TestCompileHalfDefault , model = models .resnet18 (pretrained = True )))
514
515
suite .addTest (TestPTtoTRTtoPT .parametrize (TestPTtoTRTtoPT , model = models .resnet18 (pretrained = True )))
You can’t perform that action at this time.
0 commit comments