forked from mindspore-Ecosystem/mindspore
!9772 modify yolov3-resnet18 test case
From: @chengxb7532 Reviewed-by: @c_34,@linqingke Signed-off-by: @c_34
This commit is contained in:
commit
304211e2b6
|
@ -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 = [6600, 4200, 2700]
|
||||
expect_loss_value = [6800, 4200, 2700]
|
||||
print("loss value: {}".format(loss_value))
|
||||
assert loss_value[0] < expect_loss_value[0]
|
||||
assert loss_value[1] < expect_loss_value[1]
|
||||
|
|
Loading…
Reference in New Issue