!18554 repair mobilenetv2 & inceptionv3

Merge pull request !18554 from huchunmei/repair2
This commit is contained in:
i-robot 2021-06-21 19:24:13 +08:00 committed by Gitee
commit eb93d8e39a
5 changed files with 3 additions and 7 deletions

View File

@ -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()

View File

@ -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":

View File

@ -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

View File

@ -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"

View File

@ -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