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