Skip to content

GitLab

Unverified Commit dfd853d3 authored by J-shang's avatar J-shang Committed by GitHub
Browse files

remove vgg16 from test (#4007)

parent e223988a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/ut/sdk/test_model_speedup.py
View file @ dfd853d3
...@@ -349,7 +349,7 @@ class SpeedupTestCase(TestCase): ...@@ -349,7 +349,7 @@ class SpeedupTestCase(TestCase):
self.speedup_integration(model_list) self.speedup_integration(model_list)
def test_speedup_integration_big(self): def test_speedup_integration_big(self):
model_list = ['vgg11', 'vgg16', 'resnet34', 'squeezenet1_1', model_list = ['vgg11', 'resnet34', 'squeezenet1_1',
'densenet121', 'resnet50', 'wide_resnet50_2'] 'densenet121', 'resnet50', 'wide_resnet50_2']
mem_info = psutil.virtual_memory() mem_info = psutil.virtual_memory()
ava_gb = mem_info.available/1024.0/1024/1024 ava_gb = mem_info.available/1024.0/1024/1024
...@@ -405,7 +405,7 @@ class SpeedupTestCase(TestCase): ...@@ -405,7 +405,7 @@ class SpeedupTestCase(TestCase):
if speedup_cfg is None: if speedup_cfg is None:
speedup_cfg = {} speedup_cfg = {}
ms = ModelSpeedup(speedup_model, data, ms = ModelSpeedup(speedup_model, data,
MASK_FILE, confidence=8, **speedup_cfg) MASK_FILE, confidence=6, **speedup_cfg)
ms.speedup_model() ms.speedup_model()
speedup_model.eval() speedup_model.eval()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment