Browse Source

change acc threshold of test_resnet_imagenet

tags/v0.5.0-beta
zhaoting 5 years ago
parent
commit
c083089e7e
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      tests/st/networks/models/resnet50/test_resnet50_imagenet.py

+ 4
- 4
tests/st/networks/models/resnet50/test_resnet50_imagenet.py View File

@@ -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

Loading…
Cancel
Save