mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!5828 revert ci pass condition. 

Merge pull request !5828 from anzhengqi/fix-ci-condition
This commit is contained in:
mindspore-ci-bot 2020-09-08 09:14:32 +08:00 committed by Gitee
parent 5155120ffb 38b5d91ede
commit 89c76840ee
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/st/model_zoo_tests/yolov3/test_yolov3.py
View File

@ -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.")