!18554 repair mobilenetv2 & inceptionv3
Merge pull request !18554 from huchunmei/repair2
This commit is contained in:
commit
eb93d8e39a
|
@ -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()
|
||||
|
|
|
@ -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":
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue