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