浏览代码

!11253 modify yolov3-resnet18 test case

From: @chengxb7532
Reviewed-by: @wuxuejian,@c_34
Signed-off-by: @c_34
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 年前
父节点
当前提交
26ebec42ae
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. +1
    -1
      tests/st/model_zoo_tests/yolov3/test_yolov3.py

+ 1
- 1
tests/st/model_zoo_tests/yolov3/test_yolov3.py 查看文件

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


正在加载...
取消
保存