diff --git a/model_zoo/official/cv/inceptionv4/default_config.yaml b/model_zoo/official/cv/inceptionv4/default_config.yaml index eee0fde951d..047ae129a1c 100644 --- a/model_zoo/official/cv/inceptionv4/default_config.yaml +++ b/model_zoo/official/cv/inceptionv4/default_config.yaml @@ -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" diff --git a/model_zoo/official/cv/inceptionv4/default_config_cpu.yaml b/model_zoo/official/cv/inceptionv4/default_config_cpu.yaml index e6f403b2230..714af39d769 100644 --- a/model_zoo/official/cv/inceptionv4/default_config_cpu.yaml +++ b/model_zoo/official/cv/inceptionv4/default_config_cpu.yaml @@ -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 diff --git a/model_zoo/official/cv/inceptionv4/default_config_gpu.yaml b/model_zoo/official/cv/inceptionv4/default_config_gpu.yaml index e53162651d1..b9b7a551065 100644 --- a/model_zoo/official/cv/inceptionv4/default_config_gpu.yaml +++ b/model_zoo/official/cv/inceptionv4/default_config_gpu.yaml @@ -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 diff --git a/model_zoo/official/cv/inceptionv4/eval.py b/model_zoo/official/cv/inceptionv4/eval.py index d83836adfe0..55355c7b301 100644 --- a/model_zoo/official/cv/inceptionv4/eval.py +++ b/model_zoo/official/cv/inceptionv4/eval.py @@ -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 = {