xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!17016 inceptionv4 update 

From: @huchunmei
Reviewed-by: @c_34,@wuxuejian
Signed-off-by: @c_34
This commit is contained in:
mindspore-ci-bot 2021-06-01 10:18:58 +08:00 committed by Gitee
parent 98af55560d c36394d208
commit debd27a9f2
4 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
model_zoo/official/cv/inceptionv4/default_config.yaml
View File

@ -12,8 +12,8 @@ enable_profiling: False
# ==============================================================================
dataset_path: "/cache/data"
ckpt_path: '/cache/data/'
checkpoint_path: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt'
ckpt_file: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt'
checkpoint_path: './inceptionv4/inceptionv4-train-250_1251.ckpt'
ckpt_file: '/cache/data/ImageNet_Original/inceptionv4/inceptionv4-train-250_1251.ckpt'
resume: ''
is_distributed: False
device_id: 0
@ -23,6 +23,10 @@ file_format: 'MINDIR'
width: 299
height: 299
modelarts_dataset_unzip_name: 'ImageNet_Original'
need_modelarts_dataset_unzip: True
ckpt_save_dir: './ckpt/'
# fasterrcnn_export
result_path: '' # "result file path"
label_file: '' # "label file"

8
model_zoo/official/cv/inceptionv4/default_config_cpu.yaml
View File

@ -12,8 +12,8 @@ enable_profiling: False
# ==============================================================================
dataset_path: "/cache/data"
ckpt_path: '/cache/data/'
checkpoint: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt'
ckpt_file: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt'
checkpoint_path: './inceptionv4/inceptionv4-train-250_1251.ckpt'
ckpt_file: '/cache/data/ImageNet_Original/inceptionv4/inceptionv4-train-250_1251.ckpt'
resume: ''
is_distributed: False
device_id: 0
@ -23,6 +23,10 @@ file_format: 'AIR'
width: 299
height: 299
modelarts_dataset_unzip_name: 'ImageNet_Original'
need_modelarts_dataset_unzip: True
ckpt_save_dir: './ckpt/'
# Training options
batch_size: 128
epoch_size: 250

8
model_zoo/official/cv/inceptionv4/default_config_gpu.yaml
View File

@ -12,8 +12,8 @@ enable_profiling: False
# ==============================================================================
dataset_path: "/cache/data"
ckpt_path: '/cache/data/'
checkpoint: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt'
ckpt_file: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt'
checkpoint_path: './inceptionv4/inceptionv4-train-250_1251.ckpt'
ckpt_file: '/cache/data/ImageNet_Original/inceptionv4/inceptionv4-train-250_1251.ckpt'
resume: ''
is_distributed: False
device_id: 0
@ -23,6 +23,10 @@ file_format: 'AIR'
width: 299
height: 299
modelarts_dataset_unzip_name: 'ImageNet_Original'
need_modelarts_dataset_unzip: True
ckpt_save_dir: './ckpt/'
# Training options
is_save_on_master: False

1
model_zoo/official/cv/inceptionv4/eval.py
View File

@ -83,6 +83,7 @@ def modelarts_process():
print('#' * 200, os.listdir(os.path.join(config.data_path, config.modelarts_dataset_unzip_name)))
config.dataset_path = os.path.join(config.data_path, config.modelarts_dataset_unzip_name)
config.checkpoint_path = os.path.join(config.dataset_path, config.checkpoint_path)
DS_DICT = {