From 38b5d91edecb0ed617ab44b6e31b034a87af94b0 Mon Sep 17 00:00:00 2001 From: anzhengqi Date: Mon, 7 Sep 2020 11:02:01 +0800 Subject: [PATCH] fix ci pass condition --- tests/st/model_zoo_tests/yolov3/test_yolov3.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/st/model_zoo_tests/yolov3/test_yolov3.py b/tests/st/model_zoo_tests/yolov3/test_yolov3.py index f95e9130967..ed45f02a987 100644 --- a/tests/st/model_zoo_tests/yolov3/test_yolov3.py +++ b/tests/st/model_zoo_tests/yolov3/test_yolov3.py @@ -134,7 +134,8 @@ def test_yolov3(): model = Model(net) print("Start train YOLOv3, the first epoch will be slower because of the graph compilation.") - model.train(epoch_size, dataset, callbacks=callback, dataset_sink_mode=True) + model.train(epoch_size, dataset, callbacks=callback, dataset_sink_mode=True, + sink_size=dataset.get_dataset_size()) # assertion occurs while the loss value, overflow state or loss_scale value is wrong loss_value = np.array(model_callback.loss_list) @@ -145,12 +146,12 @@ def test_yolov3(): assert loss_value[2] < expect_loss_value[2] epoch_mseconds = np.array(time_monitor_callback.epoch_mseconds_list)[2] - expect_epoch_mseconds = 2000 + expect_epoch_mseconds = 950 print("epoch mseconds: {}".format(epoch_mseconds)) assert epoch_mseconds <= expect_epoch_mseconds per_step_mseconds = np.array(time_monitor_callback.per_step_mseconds_list)[2] - expect_per_step_mseconds = 220 + expect_per_step_mseconds = 110 print("per step mseconds: {}".format(per_step_mseconds)) assert per_step_mseconds <= expect_per_step_mseconds print("yolov3 test case passed.")