remove vgg16 from test (#4007)

dfd853d3 · J-shang · GitHub · e223988a · dfd853d3
Unverified Commit dfd853d3 authored Aug 03, 2021 by J-shang Committed by GitHub Aug 03, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

test/ut/sdk/test_model_speedup.py test/ut/sdk/test_model_speedup.py +2 -2

No files found.
--- a/test/ut/sdk/test_model_speedup.py
+++ b/test/ut/sdk/test_model_speedup.py
@@ -349,7 +349,7 @@ class SpeedupTestCase(TestCase):
        self.speedup_integration(model_list)

    def test_speedup_integration_big(self):
-        model_list = ['vgg11', 'vgg16', 'resnet34', 'squeezenet1_1',
+        model_list = ['vgg11', 'resnet34', 'squeezenet1_1',
                      'densenet121', 'resnet50', 'wide_resnet50_2']
        mem_info = psutil.virtual_memory()
        ava_gb = mem_info.available/1024.0/1024/1024
@@ -405,7 +405,7 @@ class SpeedupTestCase(TestCase):
                if speedup_cfg is None:
                    speedup_cfg = {}
                ms = ModelSpeedup(speedup_model, data,
-                                  MASK_FILE, confidence=8, **speedup_cfg)
+                                  MASK_FILE, confidence=6, **speedup_cfg)
                ms.speedup_model()

                speedup_model.eval()