p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modify yolov3 test case

This commit is contained in:
chengxianbin 2021-01-14 09:37:26 +08:00
parent ee1c80abc6
commit 5afa307d7e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/model_zoo_tests/yolov3/test_yolov3.py
View File

@ -139,7 +139,7 @@ def test_yolov3():
# assertion occurs while the loss value, overflow state or loss_scale value is wrong
loss_value = np.array(model_callback.loss_list)
expect_loss_value = [6800, 4200, 2700]
expect_loss_value = [6850, 4250, 2750]
print("loss value: {}".format(loss_value))
assert loss_value[0] < expect_loss_value[0]
assert loss_value[1] < expect_loss_value[1]