!9772 modify yolov3-resnet18 test case

From: @chengxb7532
Reviewed-by: @c_34,@linqingke
Signed-off-by: @c_34
pull/9772/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 304211e2b6

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

Loading…
Cancel
Save