提交 52d5cc81 编写于 作者: S shippingwang

fix

上级 9eed230f
mode: 'train' mode: 'train'
ARCHITECTURE: ARCHITECTURE:
name: 'ResNet50_vd' name: "MobileNetV1"
pretrained_model: "" pretrained_model: ""
model_save_dir: "./output/" model_save_dir: "./output/"
...@@ -9,18 +9,18 @@ total_images: 1281167 ...@@ -9,18 +9,18 @@ total_images: 1281167
save_interval: 1 save_interval: 1
validate: True validate: True
valid_interval: 1 valid_interval: 1
epochs: 200 epochs: 120
topk: 5 topk: 5
image_shape: [3, 224, 224] image_shape: [3, 224, 224]
use_mix: True use_dali: True
ls_epsilon: 0.1
use_dali: True
LEARNING_RATE: LEARNING_RATE:
function: 'Cosine' function: 'Piecewise'
params: params:
lr: 0.1 lr: 0.1
decay_epochs: [30, 60, 90]
gamma: 0.1
OPTIMIZER: OPTIMIZER:
function: 'Momentum' function: 'Momentum'
...@@ -28,7 +28,7 @@ OPTIMIZER: ...@@ -28,7 +28,7 @@ OPTIMIZER:
momentum: 0.9 momentum: 0.9
regularizer: regularizer:
function: 'L2' function: 'L2'
factor: 0.000070 factor: 0.00003
TRAIN: TRAIN:
batch_size: 256 batch_size: 256
...@@ -36,24 +36,6 @@ TRAIN: ...@@ -36,24 +36,6 @@ TRAIN:
file_list: "./dataset/ILSVRC2012/train_list.txt" file_list: "./dataset/ILSVRC2012/train_list.txt"
data_dir: "./dataset/ILSVRC2012/" data_dir: "./dataset/ILSVRC2012/"
shuffle_seed: 0 shuffle_seed: 0
transforms:
- DecodeImage:
to_rgb: True
to_np: False
channel_first: False
- RandCropImage:
size: 224
- RandFlipImage:
flip_code: 1
- NormalizeImage:
scale: 1./255.
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: ''
- ToCHWImage:
mix:
- MixupOperator:
alpha: 0.2
VALID: VALID:
batch_size: 64 batch_size: 64
...@@ -61,18 +43,3 @@ VALID: ...@@ -61,18 +43,3 @@ VALID:
file_list: "./dataset/ILSVRC2012/val_list.txt" file_list: "./dataset/ILSVRC2012/val_list.txt"
data_dir: "./dataset/ILSVRC2012/" data_dir: "./dataset/ILSVRC2012/"
shuffle_seed: 0 shuffle_seed: 0
transforms:
- DecodeImage:
to_rgb: True
to_np: False
channel_first: False
- ResizeImage:
resize_short: 256
- CropImage:
size: 224
- NormalizeImage:
scale: 1.0/255.0
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: ''
- ToCHWImage:
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册