From c083089e7e9e510155e7cfe28830674457a7725b Mon Sep 17 00:00:00 2001 From: zhaoting Date: Wed, 17 Jun 2020 10:36:02 +0800 Subject: [PATCH] change acc threshold of test_resnet_imagenet --- .../st/networks/models/resnet50/test_resnet50_imagenet.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/st/networks/models/resnet50/test_resnet50_imagenet.py b/tests/st/networks/models/resnet50/test_resnet50_imagenet.py index 6b9b9bd7da..a322a6c883 100644 --- a/tests/st/networks/models/resnet50/test_resnet50_imagenet.py +++ b/tests/st/networks/models/resnet50/test_resnet50_imagenet.py @@ -365,8 +365,8 @@ def test_resnet_and_resnet_thor_imagenet_4p(): for i in range(device_num): os.system("rm -rf " + str(i)) print("End training...") - assert acc > 0.13 - assert cost < 21 + assert acc > 0.15 + assert cost < 20 # THOR thor_acc = 0.0 @@ -381,5 +381,5 @@ def test_resnet_and_resnet_thor_imagenet_4p(): for i in range(4, device_num + 4): os.system("rm -rf " + str(i)) print("End training...") - assert thor_acc > 0.22 - assert thor_cost < 22 + assert thor_acc > 0.25 + assert thor_cost < 21