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

change googlenet default export file_format

This commit is contained in:
caojian05 2021-05-21 15:49:21 +08:00
parent 4dfc356473
commit d42e6db078
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/googlenet/cifar10_config.yaml
View File

@ -36,7 +36,7 @@ ckpt_save_dir: "./ckpt/"
# export option
ckpt_file: ""
file_name: "googlenet"
file_format: "AIR"
file_format: "MINDIR"
#batch_size: 1
---

4
model_zoo/official/cv/googlenet/imagenet_config.yaml
View File

@ -50,7 +50,7 @@ use_label_smooth: True
# export option
ckpt_file: ""
file_name: "googlenet"
file_format: "AIR"
file_format: "MINDIR"
#batch_size: 1
---
@ -64,4 +64,4 @@ output_path: "The location of the output file."
device_target: 'Target device type'
enable_profiling: 'Whether enable profiling while training, default: False'
file_format: "choices in ['AIR', 'ONNX', 'MINDIR']"
file_format: "choices in ['AIR', 'ONNX', 'MINDIR']"