diff --git a/model_zoo/official/cv/inceptionv3/eval.py b/model_zoo/official/cv/inceptionv3/eval.py index 6ad6245c9a0..35e0a436b08 100644 --- a/model_zoo/official/cv/inceptionv3/eval.py +++ b/model_zoo/official/cv/inceptionv3/eval.py @@ -118,5 +118,4 @@ def eval_inceptionv3(): print("metric: ", metrics) if __name__ == '__main__': - config.dataset_path = os.path.join(config.dataset_path, 'validation_preprocess') eval_inceptionv3() diff --git a/model_zoo/official/cv/inceptionv3/train.py b/model_zoo/official/cv/inceptionv3/train.py index 857b59fa1e7..c2fc0adc2ed 100644 --- a/model_zoo/official/cv/inceptionv3/train.py +++ b/model_zoo/official/cv/inceptionv3/train.py @@ -103,8 +103,6 @@ def modelarts_pre_process(): @moxing_wrapper(pre_process=modelarts_pre_process) def train_inceptionv3(): print(config) - - config.dataset_path = os.path.join(config.dataset_path, 'train') create_dataset = DS_DICT[config.ds_type] if config.platform == "GPU": diff --git a/model_zoo/official/cv/mobilenetv2/default_config.yaml b/model_zoo/official/cv/mobilenetv2/default_config.yaml index 66f8f383851..463df437f6b 100644 --- a/model_zoo/official/cv/mobilenetv2/default_config.yaml +++ b/model_zoo/official/cv/mobilenetv2/default_config.yaml @@ -24,7 +24,7 @@ lr_init: 0.00 lr_end: 0.00 lr_max: 0.4 momentum: 0.9 -weight_decay: 0.00001 # 4e-5 +weight_decay: 0.00004 # 4e-5 label_smooth: 0.1 loss_scale: 1024 save_checkpoint: True diff --git a/model_zoo/official/cv/mobilenetv2/default_config_cpu.yaml b/model_zoo/official/cv/mobilenetv2/default_config_cpu.yaml index 360539e6feb..770f7cbe7a7 100644 --- a/model_zoo/official/cv/mobilenetv2/default_config_cpu.yaml +++ b/model_zoo/official/cv/mobilenetv2/default_config_cpu.yaml @@ -24,7 +24,7 @@ lr_init: .0 lr_end: 0.03 lr_max: 0.03 momentum: 0.9 -weight_decay: 0.00001 # 4e-5 +weight_decay: 0.00004 # 4e-5 label_smooth: 0.1 loss_scale: 1024 save_checkpoint: True @@ -34,7 +34,6 @@ save_checkpoint_path: "./" platform: 'CPU' run_distribute: False activation: "Softmax" -run_distribute: False # Image classification trian. train_parse_args():return train_args dataset_path: "/cache/data" diff --git a/model_zoo/official/cv/mobilenetv2/default_config_gpu.yaml b/model_zoo/official/cv/mobilenetv2/default_config_gpu.yaml index 50d7a4e3e6c..d49afa29685 100644 --- a/model_zoo/official/cv/mobilenetv2/default_config_gpu.yaml +++ b/model_zoo/official/cv/mobilenetv2/default_config_gpu.yaml @@ -24,7 +24,7 @@ lr_init: .0 lr_end: .0 lr_max: 0.8 momentum: 0.9 -weight_decay: 0.00001 # 4e-5 +weight_decay: 0.00004 # 4e-5 label_smooth: 0.1 loss_scale: 1024 save_checkpoint: True