Skip to content
Snippets Groups Projects
Commit b31fe07f authored by zhangxiaoxiao's avatar zhangxiaoxiao
Browse files

fix issue dpn/frcnn/d3p

parent 7f8f3f89
No related branches found
No related tags found
No related merge requests found
......@@ -318,10 +318,10 @@ DPN evaluate success!
### [Export MindIR](#contents)
```shell
python export.py --config_path [CONFIG_PATH] --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT]
python export.py --config_path [CONFIG_PATH] --checkpoint_path [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT]
```
The ckpt_file parameter is required,
The `checkpoint_path` parameter is required,
`FILE_FORMAT` should be in ["AIR", "MINDIR"]
- Export MindIR on Modelarts
......@@ -356,7 +356,7 @@ bash run_infer_310.sh [MINDIR_PATH] [DATASET_PATH] [DEVICE_ID]
```
- `MINDIR_PATH` specifies path of used "MINDIR" OR "AIR" model.
- `DATASET_PATH` specifies path of cifar10 datasets
- `DATASET_PATH` specifies path of imagenet2012 datasets
- `DEVICE_ID` is optional, default value is 0.
### [Result](#contents)
......
......@@ -202,7 +202,7 @@ def train_fasterrcnn():
cb += [eval_cb]
model = Model(net)
model.train(config.epoch_size, dataset, callbacks=cb, dataset_sink_mode=False)
model.train(config.epoch_size, dataset, callbacks=cb)
if __name__ == '__main__':
......
......@@ -42,7 +42,7 @@ cd ${eval_path} || exit
python ./eval.py --data_root=$DATA_ROOT \
--data_lst=$DATA_LST \
--batch_size=16 \
--batch_size=8 \
--crop_size=513 \
--ignore_label=255 \
--num_classes=21 \
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment