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

!21677 fix bug in model_zoo/cv/research/centernet_resnet50_v1 

Merge pull request !21677 from zhongkaidi/cm_master
This commit is contained in:
i-robot 2021-08-12 07:09:31 +00:00 committed by Gitee
parent 92205dafbd be96a13f89
commit dd7c7dcb83
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/research/cv/centernet_resnet50_v1/readme.md
View File

@ -390,7 +390,7 @@ overall performance on coco2017 validation dataset
### Convert ### Convert
If you want to infer the network on Ascend 310, you should convert the model to AIR: If you want to infer the network on Ascend 310, you should convert the model to MINDIR. What you need to do before is to specify the `ckpt_file` that needs to be converted in the `export_config` section of the `src/config.py` file.
```python ```python
python export.py [DEVICE_ID] python export.py [DEVICE_ID]

2
model_zoo/research/cv/centernet_resnet50_v1/scripts/ascend_distributed_launcher/hyper_parameter_config.ini
View File

@ -5,7 +5,7 @@ enable_save_ckpt=true
do_shuffle=true do_shuffle=true
enable_data_sink=true enable_data_sink=true
data_sink_steps=-1 data_sink_steps=-1
save_checkpoint_path=/root/centernet_50/model_zoo/1.0 save_checkpoint_path=./
save_checkpoint_steps=4580 save_checkpoint_steps=4580
save_checkpoint_num=30 save_checkpoint_num=30
mindrecord_prefix="coco_det.train.mind" mindrecord_prefix="coco_det.train.mind"