diff --git a/tests/st/quantization/resnet50_quant/test_resnet50_quant.py b/tests/st/quantization/resnet50_quant/test_resnet50_quant.py index 8b5d30add4..113194d30b 100755 --- a/tests/st/quantization/resnet50_quant/test_resnet50_quant.py +++ b/tests/st/quantization/resnet50_quant/test_resnet50_quant.py @@ -120,7 +120,7 @@ def test_resnet50_quant(): dataset_sink_mode=False) print("============== End Training ==============") - expect_avg_step_loss = 2.40 + expect_avg_step_loss = 2.50 avg_step_loss = np.mean(np.array(monitor.losses)) print("average step loss:{}".format(avg_step_loss))