p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!20687 ssd_mobilenet st bug fix 

Merge pull request !20687 from 张晓晓/master
This commit is contained in:
i-robot 2021-07-22 06:09:47 +00:00 committed by Gitee
parent 347b0004ae f4e02bfe1f
commit e1e7d491d9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/model_zoo_tests/ssd_mobilenet_fpn/test_ssd_modelnet_fpn.py
View File

@ -33,7 +33,7 @@ def test_SSD_mobilenet_v1_fpn_coco2017():
old_list = ["/cache/data", "MindRecord_COCO", "coco_ori", "/ckpt/mobilenet_v1.ckpt"]
new_list = [os.path.join(utils.data_root, "coco/coco2017"), "mindrecord_train/ssd_mindrecord", ".",
os.path.join(utils.ckpt_root, "ssd_mobilenet_v1/mobilenet-v1.ckpt")]
utils.exec_sed_command(old_list, new_list, os.path.join(cur_model_path, "ssd_mobilenet_v1_fpn_config.yaml"))
utils.exec_sed_command(old_list, new_list, os.path.join(cur_model_path, "config/ssd_mobilenet_v1_fpn_config.yaml"))
old_list = ["config.epoch_size", "dataset_sink_mode=dataset_sink_mode"]
new_list = ["5", "dataset_sink_mode=dataset_sink_mode, sink_size=20"]
utils.exec_sed_command(old_list, new_list, os.path.join(cur_model_path, "train.py"))