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

clould

This commit is contained in:
huchunmei 2021-05-25 15:28:51 +08:00
parent e513786709
commit c36394d208
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" dataset_path: "/cache/data"
ckpt_path: '/cache/data/' ckpt_path: '/cache/data/'
checkpoint_path: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt' checkpoint_path: './inceptionv4/inceptionv4-train-250_1251.ckpt'
ckpt_file: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt' ckpt_file: '/cache/data/ImageNet_Original/inceptionv4/inceptionv4-train-250_1251.ckpt'
resume: '' resume: ''
is_distributed: False is_distributed: False
device_id: 0 device_id: 0
@ -23,6 +23,10 @@ file_format: 'MINDIR'
width: 299 width: 299
height: 299 height: 299
modelarts_dataset_unzip_name: 'ImageNet_Original'
need_modelarts_dataset_unzip: True
ckpt_save_dir: './ckpt/'
# fasterrcnn_export # fasterrcnn_export
result_path: '' # "result file path" result_path: '' # "result file path"
label_file: '' # "label file" 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" dataset_path: "/cache/data"
ckpt_path: '/cache/data/' ckpt_path: '/cache/data/'
checkpoint: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt' checkpoint_path: './inceptionv4/inceptionv4-train-250_1251.ckpt'
ckpt_file: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt' ckpt_file: '/cache/data/ImageNet_Original/inceptionv4/inceptionv4-train-250_1251.ckpt'
resume: '' resume: ''
is_distributed: False is_distributed: False
device_id: 0 device_id: 0
@ -23,6 +23,10 @@ file_format: 'AIR'
width: 299 width: 299
height: 299 height: 299
modelarts_dataset_unzip_name: 'ImageNet_Original'
need_modelarts_dataset_unzip: True
ckpt_save_dir: './ckpt/'
# Training options # Training options
batch_size: 128 batch_size: 128
epoch_size: 250 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" dataset_path: "/cache/data"
ckpt_path: '/cache/data/' ckpt_path: '/cache/data/'
checkpoint: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt' checkpoint_path: './inceptionv4/inceptionv4-train-250_1251.ckpt'
ckpt_file: '/cache/data/inceptionv3/inceptionv3-rank3_1-247_1251.ckpt' ckpt_file: '/cache/data/ImageNet_Original/inceptionv4/inceptionv4-train-250_1251.ckpt'
resume: '' resume: ''
is_distributed: False is_distributed: False
device_id: 0 device_id: 0
@ -23,6 +23,10 @@ file_format: 'AIR'
width: 299 width: 299
height: 299 height: 299
modelarts_dataset_unzip_name: 'ImageNet_Original'
need_modelarts_dataset_unzip: True
ckpt_save_dir: './ckpt/'
# Training options # Training options
is_save_on_master: False 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))) 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.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 = { DS_DICT = {